LeetCode刷题题解：2029. 石子游戏 IX

2029. 石子游戏 IX

题目： Alice 和 Bob 再次设计了一款新的石子游戏。现有一行 n 个石子，每个石子都有一个关联的数字表示它的价值。给你一个整数数组 stones ，其中 stones[i] 是第 i 个石子的价值。

Alice 和 Bob 轮流进行自己的回合，Alice 先手。每一回合，玩家需要从 stones 中移除任一石子。

• 如果玩家移除石子后，导致 所有已移除石子 的价值 总和 可以被 3 整除，那么该玩家就 输掉游戏 。
• 如果不满足上一条，且移除后没有任何剩余的石子，那么 Bob 将会直接获胜（即便是在 Alice 的回合）。

假设两位玩家均采用 最佳 决策。如果 Alice 获胜，返回 true ；如果 Bob 获胜，返回 false 。

示例1：

输入：stones = [2,1]
输出：true
解释：游戏进行如下：

• 回合 1：Alice 可以移除任意一个石子。
• 回合 2：Bob 移除剩下的石子。
  已移除的石子的值总和为 1 + 2 = 3 且可以被 3 整除。因此，Bob 输，Alice 获胜。

示例2：

输入：stones = [2]
输出：false
解释：Alice 会移除唯一一个石子，已移除石子的值总和为 2 。
由于所有石子都已移除，且值总和无法被 3 整除，Bob 获胜。

示例3：

输入：stones = [5,1,2,4,3]
输出：false
解释：Bob 总会获胜。其中一种可能的游戏进行方式如下：
– 回合 1：Alice 可以移除值为 1 的第 2 个石子。已移除石子值总和为 1 。
– 回合 2：Bob 可以移除值为 3 的第 5 个石子。已移除石子值总和为 = 1 + 3 = 4 。
– 回合 3：Alices 可以移除值为 4 的第 4 个石子。已移除石子值总和为 = 1 + 3 + 4 = 8 。
– 回合 4：Bob 可以移除值为 2 的第 3 个石子。已移除石子值总和为 = 1 + 3 + 4 + 2 = 10.
– 回合 5：Alice 可以移除值为 5 的第 1 个石子。已移除石子值总和为 = 1 + 3 + 4 + 2 + 5 = 15.
Alice 输掉游戏，因为已移除石子值总和（15）可以被 3 整除，Bob 获胜。

解题思路：
我们可以按照石子价值除以 3 的余数分成三类，并统计相应数量。某个回合开始前，已移除的石子总和状态为 x（共三种，分别为除以 3 余数为 0、1 和 2，其中当状态为 0，且非首个回合时，说明凑成 3 的倍数，游戏结束），剩余石子价值除以 3 的余数 s 分别为 0、1 和 2。
首先如果当前 x = 1 时，不能选择 s = 2 的石子，否则会导致凑成总和为 3 的倍数而失败；同理 x = 2 时，不能选择 s = 1 的石子；而选择 s = 0 的数字，不会改变 x 的状态，可看做换手操作。同时成对的 s = 0 的等价于没有 s = 0 的石子（双方只需要轮流选完这些 s = 0 的石子，最终会回到先手最开始的局面）；而选择与 x 相同的 s 会导致 x 改变（即 x = 1 时，选择 s = 1 的石子，会导致 x = 2；而 x = 2 时，选 s = 2 的石子，会导致 x = 1）。

所以，经过分情况讨论，得出Alice获胜的情况只有如下两种：

1. s=0 的石子数量为偶数时，s=1 和 s=2 的石子数量均不为 0
2. s=0 的石子数量为奇数时，两者数量差超过 2

Java代码：

class Solution {
    public boolean stoneGameIX(int[] stones) {
        int[] cnts = new int[3];
        for (int i : stones) cnts[i % 3]++;
        return cnts[0] % 2 == 0 ? !(cnts[1] == 0 || cnts[2] == 0) : !(Math.abs(cnts[1] - cnts[2]) <= 2);
    }
}

Go代码：

func stoneGameIX(stones []int) bool {
    cnts := [3]int{}
    for _, i := range stones {
        cnts[i % 3]++
    }
    if cnts[0] % 2 == 0 {
        return !(cnts[1] == 0 || cnts[2] == 0)
    } else {
        return cnts[1] - cnts[2] > 2 || cnts[2] -cnts[1] > 2
    }
}

复杂度分析：

• 时间复杂度：O(n)
• 空间复杂度：O(1)