leetcode/1717.cpp at master · Yashbhatt7/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#include<iostream>
#include<string>

int maximumGain(std::string s, int x, int y) {
    int score = 0;

    for(int i = 0; i < s.size(); ++i) {
        if(y > x) {
            auto idx = s.find("ba", i);
            if(idx != std::string::npos) {
                s.erase(s.begin() + idx, s.begin() + (idx + 2));
                score += y;
                i = i - 1;
            }
            else {
                auto idx1 = s.find("ab", i);
                if(idx1 != std::string::npos) {
                    s.erase(s.begin() + idx1, s.begin() + (idx1 + 2));
                    score += x;
                    i = i - 1;
                }
            }
        } else {
            auto idx3 = s.find("ab", i);
            if(idx3 != std::string::npos) {
                auto ss = s.erase(s.begin() + idx3, s.begin() + (idx3 + 2));
                score += x;
                i = i - 1;
            }
            else {
                auto idx2 = s.find("ba", i);
                if(idx2 != std::string::npos) {
                    s.erase(s.begin() + idx2, s.begin() + (idx2 + 2));
                    score += y;
                    i = i - 1;
                }
            }
        }
    }

    return score;
}

int main() {
    std::string s = "aabbaaxybbaabb";
    int x = 5;
    int y = 4;

    std::cout << maximumGain(s, x, y);
    // maximumGain(s, x, y);

    std::cin.get();
}