20.12.11 649. Dota2 参议院

题目

Dota2 的世界里有两个阵营：Radiant(天辉)和 Dire(夜魇)
Dota2 参议院由来自两派的参议员组成。现在参议院希望对一个 Dota2 游戏里的改变作出决定。他们以一个基于轮为过程的投票进行。在每一轮中，每一位参议员都可以行使两项权利中的一项：
禁止一名参议员的权利：参议员可以让另一位参议员在这一轮和随后的几轮中丧失所有的权利。
宣布胜利：如果参议员发现有权利投票的参议员都是同一个阵营的，他可以宣布胜利并决定在游戏中的有关变化。
给定一个字符串代表每个参议员的阵营。字母 “R” 和 “D” 分别代表了 Radiant（天辉）和 Dire（夜魇）。然后，如果有 n 个参议员，给定字符串的大小将是 n。
以轮为基础的过程从给定顺序的第一个参议员开始到最后一个参议员结束。这一过程将持续到投票结束。所有失去权利的参议员将在过程中被跳过。
假设每一位参议员都足够聪明，会为自己的政党做出最好的策略，你需要预测哪一方最终会宣布胜利并在 Dota2 游戏中决定改变。输出应该是 Radiant 或 Dire。

示例 1：
输入："RD"
输出："Radiant"
解释：第一个参议员来自 Radiant 阵营并且他可以使用第一项权利让第二个参议员失去权力，因此第二个参议员将被跳过因为他没有任何权利。然后在第二轮的时候，第一个参议员可以宣布胜利，因为他是唯一一个有投票权的人

示例 2：
输入："RDD"
输出："Dire"
解释：
第一轮中,第一个来自 Radiant 阵营的参议员可以使用第一项权利禁止第二个参议员的权利
第二个来自 Dire 阵营的参议员会被跳过因为他的权利被禁止
第三个来自 Dire 阵营的参议员可以使用他的第一项权利禁止第一个参议员的权利
因此在第二轮只剩下第三个参议员拥有投票的权利,于是他可以宣布胜利

提示：
给定字符串的长度在 [1, 10,000] 之间.

思路

1. 敲了30分钟，太慢了。题目没理解透彻，边调试边改，假如没有lc的调试功能可能会更慢（牛客面试GG）
2. 纯模拟，用了erase就感觉自己的方法绝对不是最优的了，因为erase的时间复杂度是O(n)，不过数据量是10000，问题不大
3. 思路就是先记录各自阵营有多少个人，del是记录要禁止的人数
4. 循环输入字符串，如果当前阵营不需要del，那么它就可以使用第一项权利去进制别人；如果需要del，即在轮到它前，它被另一个阵营禁止了。同时要把它从字符串中删除，因为下一轮还需要遍历字符串
5. 看了题解，另一种思路是用两个队列，分别记录它们投票的时间

代码

class Solution {
public:
    string predictPartyVictory(string senate) {
        string ans;
        int restR = 0, restD = 0;
        for(auto s : senate){
            if(s == 'R') ++restR;
            else ++restD;
        }

        int delR = 0, delD = 0;
        while(restR && restD){
            for(int i = 0; i < senate.size(); ++i){
                if(senate[i] == 'R'){
                    if(delR){
                        --delR;
                        senate.erase(i, 1);
                        --i;
                    }
                    else{
                        ++delD;
                        --restD;
                    } 
                }
                else if(senate[i] == 'D'){
                    if(delD){
                        --delD;
                        senate.erase(i, 1);
                        --i;
                    }
                    else{
                        ++delR;
                        --restR;
                    } 
                }
                if(restD == 0 || restR == 0) break;
            }
        }

        ans = restR?"Radiant":"Dire";
        return ans;
    }
};

class Solution {
public:
    string predictPartyVictory(string senate) {
        int n = senate.size();
        string ans;
        queue<int> qR, qD;
        for(int i = 0; i < n; ++i){
            if(senate[i] == 'R') qR.push(i);
            else qD.push(i);
        }

        while(!qD.empty() && !qR.empty()){
            if(qD.front() > qR.front()) qR.push(qR.front()+n);
            else qD.push(qD.front()+n);
            qR.pop();
            qD.pop();
        }

        ans = qD.empty()?"Radiant":"Dire";

        return ans;
    }
};