题解 | 参议院投票

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 求出最终获胜帮派的名称
# @param s string字符串 
# @return string字符串
#
class Solution:
    def predictVictory(self , s: str) -> str:
        # write code here
        n = len(s)
        rq = [i for i, char in enumerate(s) if char=='R']#标记红帮/黑帮议员下标序号
        dq = [i for i, char in enumerate(s) if char=='D']

        while rq and dq:#如果红帮/黑帮议员都还有行动力（没被禁止）
            ri = rq.pop(0)#红帮/黑帮最前列的议员分别出列
            di = dq.pop(0)

            if ri<di:#谁的序号大，谁被禁止行动，同时序号小的进入下一轮
                rq.append(ri+n)
            else:
                dq.append(di+n)
        
        return 'Dark' if not rq else 'Red'#当红帮可行动议员为空，则黑帮胜，反之亦然