[LeetCode] 649. Dota2 Senate

题目描述：

 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 阵营的参议员可以使用他的第一项权利禁止第一个参议员的权利
因此在第二轮只剩下第三个参议员拥有投票的权利,于是他可以宣布胜利

解题思路：

每个有执行权力的议员优先对排在自己之后并且距离自己最近的不同阵营成员禁言，

根据该策略，将不同阵营的成员清除，循环直到只剩下一方阵营为止。

思路有了，写代码也就容易了。

第一版：

class Solution:
    def predictPartyVictory(self, senate):
        """
        :type senate: str
        :rtype: str
        """
        while 1:
            forbidden=0
            n=len(senate)
            mark=[0 for i in range(n)]
            for i in range(0,n):
                if mark[i]==0:
                    if senate[i]=='R':
                        Mark=0
                        for j in range(i+1,i+n):
                            if mark[j%n]==0 and senate[j%n]=='D':
                                mark[j%n]=1
                                forbidden= forbidden+1
                                Mark=1
                                break
                        if Mark==0:
                            return "Radiant"
                    if senate[i]=='D':
                        Mark=0
                        for j in range(i+1,i+n):
                            if mark[j%n]==0 and senate[j%n]=='R':
                                mark[j%n]=1
                                Mark=1
                                forbidden= forbidden+1
                                break
                        if Mark==0:
                            return "Dire"
            tmp=[0 for i in range(n-forbidden)] 
            j=0
            for i in range(n):
                if mark[i]==0:
                    tmp[j]=senate[i]
                    j=j+1
            senate=tmp

大部分测试都可以通过，但对于很长的就出现超出时间限制，分析时间复杂度就可知每次循环都有O(n^2)的复杂度。

第二版：

参考了该博客中的方法：http://bookshadow.com/weblog/2017/07/30/leetcode-dota2-senate/

class Solution:
    def predictPartyVictory(self, senate):
        """
        :type senate: str
        :rtype: str
        """
        label=0
        while(len(set(senate))>1):
            tmp=''
            for i in senate:
                if i=='R':
                    if label>=0:
                        tmp=tmp+'R'
                    label+=1
                if i=='D':
                    if label<=0:
                        tmp=tmp+'D'
                    label-=1
            senate=tmp
        return {'D':"Dire",'R':"Radiant"}[senate[0]]

这样每次循环只有O(n)的复杂度，了不起了不起。