LeetCode-Python-1510. 石子游戏 IV（DP)

Alice 和 Bob 两个人轮流玩一个游戏，Alice 先手。

一开始，有 n 个石子堆在一起。每个人轮流操作，正在操作的玩家可以从石子堆里拿走 任意 非零 平方数 个石子。

如果石子堆里没有石子了，则无法操作的玩家输掉游戏。

给你正整数 n ，且已知两个人都采取最优策略。如果 Alice 会赢得比赛，那么返回 True ，否则返回 False 。

 

示例 1：

输入：n = 1
输出：true
解释：Alice 拿走 1 个石子并赢得胜利，因为 Bob 无法进行任何操作。
示例 2：

输入：n = 2
输出：false
解释：Alice 只能拿走 1 个石子，然后 Bob 拿走最后一个石子并赢得胜利（2 -> 1 -> 0）。
示例 3：

输入：n = 4
输出：true
解释：n 已经是一个平方数，Alice 可以一次全拿掉 4 个石子并赢得胜利（4 -> 0）。
示例 4：

输入：n = 7
输出：false
解释：当 Bob 采取最优策略时，Alice 无法赢得比赛。
如果 Alice 一开始拿走 4 个石子， Bob 会拿走 1 个石子，然后 Alice 只能拿走 1 个石子，Bob 拿走最后一个石子并赢得胜利（7 -> 3 -> 2 -> 1 -> 0）。
如果 Alice 一开始拿走 1 个石子， Bob 会拿走 4 个石子，然后 Alice 只能拿走 1 个石子，Bob 拿走最后一个石子并赢得胜利（7 -> 6 -> 2 -> 1 -> 0）。
示例 5：

输入：n = 17
输出：false
解释：如果 Bob 采取最优策略，Alice 无法赢得胜利。
 

提示：

1 <= n <= 10^5
通过次数1,053提交次数2,577

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/stone-game-iv
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

已知 Alice 先手，且每个人每次只能拿非零完全平方数个石子，

当 n = 1时，显然 Alice 胜，

当 n = 2时，Alice 只能拿 1 个。剩下的游戏对于Bob来说，他就是上一行的Alice，所以Bob 胜，Alice 输。

当 n = 3时，Alice 只能拿 1 个。剩下的游戏对于Bob来说，他就是上一行的Alice，所以Alice 胜，Bob 输。

以此类推。。。

只要Alice能拿 k * k (k > 0)个石子，使得剩下的游戏对于Bob来说是稳输的局面，Alice就可以稳赢。

设 dp[i] = True/ False 代表 n = i 时，Alice能不能赢。

状态转移方程为：

dp[i] = (存在 dp[j] = dp[i - k * k]， 使得dp[j] 为负，其中 0 < k <= i ** 0.5)

时间复杂度：O（NlogN）

空间复杂度：O（N）

class Solution(object):
    def winnerSquareGame(self, n):
        """
        :type n: int
        :rtype: bool
        """
        dp = [0 for _ in range(n + 1)]
        dp[1] = 1
        for i in range(2, n + 1):
            for j in range(1, int(i ** 0.5 + 1)):
                if not dp[i - j * j]:
                    dp[i] = 1
                    break
        return dp[n] == 1