leetcode——【石子游戏IV（博弈类）】

题目

思路

石头可简化为价值为0，1， 2的情况，对于价值为0的石头，可做如下简化：
1. 如果0的个数为偶数，等于价值0个数为0的情况；
2. 如果0的个数为奇数，等于价值0个数为1的情况。
对于价值为1和2的石头，可以简化为两种情况：
1. A首先移除了1，那么B只能移除1，之后A只能移除2，B只能移除1；
2. A首先移除了2，那么B只能移除2，之后A只能移除1，B只能移除2。
因此可以得到算法策略：
如果第一种情况，A获胜：

• 1恰好有一个，2至少有一个；
• 1至少有两个，并且小于等于2的个数；
  即有价值1的石头，并且不能比价值2的石头多。
  B获胜：
• 价值2比价值1的石头多，若多一个或者两个，因为所有石头被移除获胜；若多的个数大于两个，因为A没有价值2的石头移除而获胜。

如果第二种情况，A获胜：
有价值2的石头，并且不能比价值1的石头多。
两种情况为或的关系，则最终策略为：

价值1和价值2的石头至少都有一个

综合价值0的石头可以得到策略：

如果价值0的个数为偶数，则A获胜条件为1和2至少有一个；如果价值0的石头个数和价值1的石头个数差值为2（在没价值0石头情况下，B获胜且不是因为所有石头都被移除）。

实现

class Solution {
public:
    bool stoneGameIX(vector<int>& stones) {
        int stone_0_count = 0, stone_1_count = 0, stone_2_count = 0;
        for (int index : stones)
            if (index % 3 == 0)
                ++stone_0_count;
            else if (index % 3 == 1)
                ++stone_1_count;
            else
                ++stone_2_count;
        if (stone_0_count % 2 == 0)
            return stone_1_count >= 1 && stone_2_count >= 1;
        return stone_1_count - stone_2_count > 2 || stone_2_count - stone_1_count > 2;
    }
};