Dota2 Senate LeetCode solution in Python

1
from collections import deque
2

3

4
class Solution:
5
    def predictPartyVictory(self, senate: str) -> str:
6
        n = len(senate)
7
        radiant = deque()
8
        dire = deque()
9
        for i, s in enumerate(senate):
10
            if s == "R":
11
                radiant.append(i)
12
            else:
13
                dire.append(i)
14
        while radiant and dire:
15
            r_idx = radiant.popleft()
16
            d_idx = dire.popleft()
17
            if r_idx < d_idx:
18
                radiant.append(r_idx + n)
19
            else:
20
                dire.append(d_idx + n)
21
        return "Radiant" if radiant else "Dire"