解题思路:
通过每一轮的投票,看哪一方最后还剩余可以投票的参议员来判定。
通过一个vis数组标志是否有权利进行下一轮投票。
r_sum 记录总共的R参议员人数, cir_numr 记录本轮可以投票的R参议员人数。
d_sum 记录总共的D参议员人数, cir_numd 记录本轮可以投票的D参议员人数。
每一轮,判定当前位置是D or R, 例如当前是D ,如果cir_numr == 0 说明没有R参议员阻止我投票,我可以阻止下一个R参议员投票权 ; 如果cir_numr != 0,那么R参议员就可以阻止我投票,我的权利就没有了,以后也没有了,但是可以投票的R参议员人数 -- , 所以 cir_numr-- ,d_sum --。
对于当前是R的情况也是一样的,直到最后哪一方的人数为0.
代码:
class Solution {
public:
string predictPartyVictory(string senate)
{
int r_sum = 0;
int d_sum = 0;
int sz = senate.size();
vector<bool> vis(sz, true);
for (auto& e : senate)
{
if (e == 'R')
{
r_sum++;
}
else
{
d_sum++;
}
}
int cir_numr = 0;
int cir_numd = 0;
while (r_sum && d_sum)
{
for (int i = 0; i < sz; ++i)//每一轮投票
{
if (vis[i] == true && senate[i] == 'D')//当前位置还可以用
{
if (cir_numr == 0)//在这个人之前,本轮参与投票的R参议员的个数为0,
{
cir_numd++;//票先保留着,遇到后面R的参议院,不让说话
}
else
{
cir_numr--;//之前的R参议院用了自己的权利,让本位置的人失去权利
d_sum--;//总的D参议院可用-1
vis[i] = false;
}
}
else if(vis[i] == true && senate[i] == 'R')
{
if (cir_numd == 0)
{
cir_numr++;
}
else
{
cir_numd--;
r_sum--; //总的D参议院可用
vis[i] = false;
}
}
}
}
if (r_sum == 0)
{
return string("Dire");
}
else
{
return string("Radiant");
}
}
};