题目描述
Dota2 议会由两个政党 Radiant 和 Dire 组成,现在议会要做出 Dota2 游戏规则的改变的决定,由两个政党按轮次投票决定,每一轮每个议员可使用以下两种权力之一:
- 禁止一个议员的权力(包括该轮次及所有后续轮次)
- 宣布胜利(如果拥有权力的议员都属于该政党)
给定一个字符串,每个代表是 ‘R’ 或者 ‘D’,分别代表每个议员所属的政党 ‘Radiant’ 或者 ‘Dire ’。如果有 N 个议员,则该字符串的长度为 N。每一轮的投票流程按给定顺序从第一个议员到最后一个议员依次进行。流程直到投票结束,没有权力投票的议员将被跳过。
预测哪一个政党会宣布胜利,‘Radiant’ 或者 ‘Dire ’ ?
Example:
Input: “RD”
Output: “Radiant”
解题思路
本题具有明显的 贪心
背景,每个议员只需禁止下一个非本党派议员即可。我首先尝试了暴力贪心,由于每次需遍历非必要的同党议员和被淘汰的议员,时间开销太大,提交超时。然后在讨论区发现了一种精妙的解法!用两个队列分别存储两个政党议员的索引,在每一轮,删掉被淘汰议员的索引,把被保留的议员的索引加 N
移至所属队列最后面。
def predictPartyVictory(self, senate):
"""
:type senate: str
:rtype: str
"""
q1, q2 = [], []
N = len(senate)
for i in range(N):
if senate[i] == 'R':
q1.append(i)
else:
q2.append(i)
while q1 and q2:
r_index, d_index = q1.pop(0), q2.pop(0)
if r_index < d_index:
q1.append(r_index + N) # smart idea to use _index + n
else:
q2.append(d_index + N)
return "Radiant" if q1 else "Dire"
该解法的巧妙之处是:分两个队列分别存储索引,并将被保留的议员索引 +N 移植队列尾部。