LeetCode649. Dota2 参议院

649. 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, 10,000] 之间.

Solution

重点是禁止下一个离他最近的对方阵营的参议员。

class Solution(object):
    def predictPartyVictory(self, senate):
        """
        :type senate: str
        :rtype: str
        """
        Senate=list(senate)
        while 'D' in Senate and 'R' in Senate:
        #就一派人了,可以宣布胜利
            l=len(Senate)
            i=0
            while i<l:
                x=Senate[i]
                if x=='R' and 'D' in Senate:
                    #禁止下一个离他最近的对方阵营的参议员(重点)
                    #注意搜索顺序
                    for j in range(i,l)+range(0,i):
                        if Senate[j]=='D':
                            break
                    del Senate[j]
                    l-=1
                elif x=='D' and 'R' in Senate:
                    for j in range(i,l)+range(0,i):
                        if Senate[j]=='R':
                            break
                    del Senate[j]
                    l-=1
                i+=1
        return 'Radiant' if Senate[0]=='R' else 'Dire'

官方题解：

由于我们总要挑选投票顺序最早的议员，因此我们可以使用两个队列 radiant 和 dire 分别按照投票顺序存储天辉方和夜魇方每一名议员的投票时间。随后我们就可以开始模拟整个投票的过程：

• 如果此时 radiant 或者 dire 为空，那么就可以宣布另一方获得胜利；

• 如果均不为空，那么比较这两个队列的首元素，就可以确定当前行使权利的是哪一名议员。如果 radiant 的首元素较小，那说明轮到天辉方的议员行使权利，其会挑选 dire 的首元素对应的那一名议员。因此，我们会将 dire 的首元素永久地弹出，并将 radiant 的首元素弹出，增加 n 之后再重新放回队列，这里 n 是给定的字符串 senate 的长度，即表示该议员会参与下一轮的投票。

为什么这里是固定地增加 n，而不是增加与当前剩余议员数量相关的一个数？
答：他本来哪个位置就在那呗。。这样相对位置肯定没问题（想当然。。）

同理，如果 dire 的首元素较小，那么会永久弹出 radiant 的首元素，剩余的处理方法也是类似的。

另外这里用了deque代替list，是一个双端队列，适合经常从两端append数据时使用，可高效实现插入和删除操作。

class Solution:
    def predictPartyVictory(self, senate: str) -> str:
        n = len(senate)
        radiant = collections.deque()
        dire = collections.deque()
        
        for i, ch in enumerate(senate):
            if ch == "R":
                radiant.append(i)
            else:
                dire.append(i)
        
        while radiant and dire:
            if radiant[0] < dire[0]:
                radiant.append(radiant[0] + n)
            else:
                dire.append(dire[0] + n)
            radiant.popleft()
            dire.popleft()
        
        return "Radiant" if radiant else "Dire"

collections.deque文档
collections.deque使用介绍