Leetcode 649(12-11-2020) Python

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] 之间.

解题思路：

使用贪心算法，可以额外开一个数组，如果之后又一个其他的阵营的，就禁了
如果之后都没有了，就禁前面的，直到只有一个阵营的人

代码：

超出时间限制版【用python写出了C的感觉】

class Solution:
    def predictPartyVictory(self, senate: str) -> str:
        l = len(senate)
        x = []
        res = [0]*2
        for i in senate:
            x.append(i) 
            if i == 'R':
                res[0] += 1
            else:
                res[1] += 1      
        # if l == 1:
        #     if senate == 'R':
        #         return "Radiant"
        #     else:
        #         return "Dire"
        while res[0] and res[1]:
            for i in range(l):
                if x[i] == 'R':
                    for k in range(i,l):
                        if x[k] == 'D':
                            x[k] = 'd'
                            res[1] -= 1
                            break
                        if k == l-1:
                            for k in range(i):
                                if x[k] == 'D':
                                    x[k] = 'd'
                                    res[1] -= 1
                                    break
                elif x[i] == 'D':
                    for k in range(i,l):
                        if x[k] == 'R':
                            x[k] = 'r'
                            res[0] -= 1
                            break
                        if k == l-1:
                            for k in range(i):
                                if x[k] == 'R':
                                    x[k] = 'r'
                                    res[0] -= 1
                                    break
        if res[0]:
            return "Radiant"
        else:
            return "Dire"

标答版：

     else:
                for k in range(i,l):
                    if senate[k] == 'R':
                        senate[k].pop()
                        break

        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"

参考链接：

1. https://leetcode-cn.com/problems/dota2-senate/solution/java-649dota2can-yi-yuan-chao-9961-by-mu-0dqh/
2. https://leetcode-cn.com/problems/dota2-senate/solution/dota2-can-yi-yuan-by-leetcode-solution-jb7l/