649. Dota2 参议院

第一次尝试 暴力

本来想统计字符串里面R和D的数目，发现想得太简单了

“DDRRR”

虽然R多，但却是D赢了，因为D先行ban了两个R，还剩一个R ban了一个D，然后剩下那个D又把R给ban了，最后5个只剩下一个D所以是D赢

第二次尝试 模拟

观察上面的投票过程，只有当队列中能够投票的元素均相同的时候才能宣布胜利，否则每个元素都会禁止离它最近的下一个敌对元素，那么我们按照这个策略来模拟

1. 使用r和d来统计当前分别能够投票的元素个数，使用flag数组标记flag[i]是否有投票权利
2. 遍历队列，直到r或d为0，执行以下操作
3. 从当前元素开始寻找最近的敌对元素，修改其flag[i]=false
4. 返回不为0的元素

class Solution:
    def predictPartyVictory(self, senate: str) -> str:
        size = len(senate)
        r,d = 0,0
        for ch in senate:
            if ch == 'R':
                r+=1
            else:
                d+=1
        flag = [True] * size
        ind = 0
        while r!=0 and d!=0:
            if flag[ind]:
                ch = senate[ind]
                j = (ind+1)%size
                while True:
                    if ch == 'R' and senate[j] == 'D' and flag[j] == True:
                        d-=1
                        break
                    if ch == 'D' and senate[j] == 'R' and flag[j] == True:
                        r-=1
                        break
                    j = (j+1)%size
                flag[j] = False
            ind = (ind+1)%size
            # print(flag)

        if r!=0:
            return 'Radiant'
        else:
            return 'Dire'

            

            
        

通过了，但是效率很低，主要是往后找元素这个过程花费时间太长了

改进

我们只遍历一轮，用r和d来表示当前拥有投票权的元素

1. 如果当前元素是R，并且d==0，那么r+=1，反之同理
2. 如果当前元素是R，但是d!=0，那么不做任何操作，反之同理
3. 遍历一轮后检查r和d的数量

但是我们每个元素在一轮决策中只能投票一次，所以上面的遍历还要改进，我们现在用两个队列r和d来存储可以投票的元素

1. 如果当前元素是R，d队列为空，那么入队r队列，否则d队列出队一个元素，存储到res字符串中等待下一轮投票，反之同理
2. 取出res字符串中的元素，重复上述遍历
3. 如果res为空，那么我们检查队列元素，返回相应答案

class Solution:
    def predictPartyVictory(self, senate: str) -> str:
        size = len(senate)
        r,d = [],[] 
        res = senate 
        while res!='':
            senate = res
            res = ''
            for ch in senate:
                if ch=='R':
                    if len(d)==0:
                        r.append('R') 
                    else:
                        res+=d.pop()
                else:
                    if len(r)==0:
                        d.append('D')
                    else:
                        res+=r.pop()
                        
        if len(r)!=0:
            return 'Radiant'
        else:
            return 'Dire'     

这次效率高了很多

标签：ch,senate,res,元素,flag,Dota2,649,参议院,size
来源： https://blog.csdn.net/hhmy77/article/details/111030309