[Leetcode] 649. Dota2 Senate

 

🗓️ Daily LeetCoding Challenge May, Day 4

 

class Solution:
    def predictPartyVictory(self, senate: str) -> str:

        A = collections.deque()
        people = [0, 0]
        bans = [0, 0]

        for person in senate:
            x = person == 'R'
            people[x] += 1
            A.append(x)
        
        while all(people):
            x = A.popleft()
            people[x] -= 1
            if bans[x]:
                bans[x] -= 1
            else:
                people[x] +=1
                bans[x^1] += 1
                A.append(x)

        return 'Radiant' if people[1] else 'Dire'