LeetCode刷题之路 Dota2 参议院

Dota2 参议院【中等】

Dota2 的世界里有两个阵营：Radiant(天辉)和 Dire(夜魇)

Dota2 参议院由来自两派的参议员组成。现在参议院希望对一个 Dota2 游戏里的改变作出决定。他们以一个基于轮为过程的投票进行。在每一轮中，每一位参议员都可以行使两项权利中的**一**项：

1. 禁止一名参议员的权利：

   参议员可以让另一位参议员在这一轮和随后的几轮中丧失所有的权利。

2. 宣布胜利：如果参议员发现有权利投票的参议员都是同一个阵营的，他可以宣布胜利并决定在游戏中的有关变化。

给定一个字符串代表每个参议员的阵营。字母“R”和“D”分别代表了Radiant(天辉)和 Dire(夜魇)。然后，如果有 n 个参议员，给定字符串的大小将是n。

以轮为基础的过程从给定顺序的第一个参议员开始到最后一个参议员结束。这一过程将持续到投票结束。所有失去权利的参议员将在过程中被跳过。

假设每一位参议员都足够聪明，会为自己的政党做出最好的策略，你需要预测哪一方最终会宣布胜利并在 Dota2 游戏中决定改变。输出应该是 Radiant 或 Dire。

示例 1:

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

示例 2:

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

注意:

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

解题思路

一看这个题目大概就能知道最优的策略就是禁掉自己后面的敌人，没有的话再禁掉前面的敌人。在这里，我们要分别记录两个阵营的位置，所以分别创建了两个队列分别存储两个阵营的每个人的位置，对senate进行遍历不仅得到了两个阵营的每个人的位置，而且也得到了senate的长度i。然后进行一个循环直到某一个阵营没有人为止，分别取出队伍最前面的两个阵营的人的位置，决定淘汰哪个人（位置靠前的阵营的人淘汰位置相对靠后的那个人），然后将暂时没被淘汰的人加到队伍后（代表位置的数值也相应改变）。最后进行判断，返回有剩余的阵营的名称。

class Solution:
    def predictPartyVictory(self, senate):
        """
        :type senate: str
        :rtype: str
        """
        from collections import deque
        R = deque()
        D = deque()
        i = 0
        for sen in senate:
            if sen == "R":
                R.append(i)
                i = i + 1
            else:
                D.append(i)
                i = i + 1
        while R and D:
            index_R = R.popleft()
            index_D = D.popleft()
            if index_R < index_D:
                R.append(index_R + i)
            else:
                D.append(index_D + i)
        if R:
            return 'Radiant'
        else:
            return 'Dire'