Dota2参议院-贪心649-python

没看答案，贪心思想。

class Solution:
    def predictPartyVictory(self, senate: str) -> str:
        '''
        贪心思想胜利策略：优先ban后面出现的敌对方，如果后面没有敌对方，ban前面出现的敌对方，直到胜利。
        '''
        sen = list(senate)

        while True:
            num_R = num_D = 0 # 当前剩余的R or D数量
            ban_R = ban_D = 0 # 当前需要ban掉的R or D数量

            for idx, ch in enumerate(sen):
                if ch == 'D':
                    if ban_D == 0: # 如果没被ban，则行使ban人权力
                        ban_R += 1
                        num_D += 1
                    else:          # 如果被ban了，则被注销为'0'，并填补一个被ban名额
                        sen[idx] = '0'
                        ban_D -= 1

                elif ch == 'R':
                    if ban_R == 0: # 如果没被ban，则行使ban人权力
                        ban_D += 1
                        num_R += 1
                    else:          # 如果被ban了，则被注销为'0'，并填补一个被ban名额
                        sen[idx] = '0'
                        ban_R -= 1
            
            if ban_D: # 如果R行使ban人权力时后面没有出现的D，则ban掉前面出现的D
                for idx, c in enumerate(sen):
                    if c == 'D':
                        sen[idx] = '0'
                        ban_D -= 1
                        num_D -= 1
                    if not ban_D:
                        break
            
            if ban_R: # 如果D行使ban人权力时后面没有出现的R，则ban掉前面出现的R
                for idx, c in enumerate(sen):
                    if c == 'R':
                        sen[idx] = '0'
                        ban_R -= 1
                        num_R -= 1
                    if not ban_R:
                        break

            # 如果有一方人员不再剩余，则宣布另一方获胜
            if not num_D:
                return 'Radiant'
            elif not num_R:
                return 'Dire'