works/solutions/leetcode/cpp/649.cpp


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
54
55
56
57
58
59
60
61
62
63
64
65
66
67

#include <string>

using std::string;

#include <queue>

class Solution
{
public:
    string predictPartyVictory(string senate)
    {
        std::queue<bool> queue;

        int r_people = 0;
        int d_people = 0;
        int r_ban = 0;
        int d_ban = 0;

        for (auto i = senate.cbegin(); i != senate.cend(); ++i)
        {
            if (*i == 'R')
            {
                r_people += 1;
                queue.push(true);
            }
            else
            {
                d_people += 1;
                queue.push(false);
            }
        }

        while (r_people && d_people)
        {
            bool is_r = queue.front();
            queue.pop();
            if (is_r)
            {
                if (d_ban)
                {
                    r_people -= 1;
                    d_ban -= 1;
                }
                else
                {
                    r_ban += 1;
                    queue.push(is_r);
                }
            }
            else
            {
                if (r_ban)
                {
                    d_people -= 1;
                    r_ban -= 1;
                }
                else
                {
                    d_ban += 1;
                    queue.push(is_r);
                }
            }
        }

        return r_people ? "Radiant" : "Dire";
    }
};