LeetCode - 649. Dota2 参议院

力扣

 

解题思路:

通过每一轮的投票，看哪一方最后还剩余可以投票的参议员来判定。

通过一个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");
		}
    }
    
};