题目
你和你的朋友,两个人一起玩 Nim 游戏:
- 桌子上有一堆石头。
- 你们轮流进行自己的回合,你作为先手。
- 每一回合,轮到的人拿掉 1 - 3 块石头。
- 拿掉最后一块石头的人就是获胜者。
假设你们每一步都是最优解。请编写一个函数,来判断你是否可以在给定石头数量为 n 的情况下赢得游戏。如果可以赢,返回 true;否则,返回 false 。
示例
示例1
输入: n = 4
输出: False
解释: 如果堆中有 4 块石头,那么你永远不会赢得比赛;因为无论你拿走 1 块、2 块 还是 3 块石头,最后一块石头总是会被你的朋友拿走。
示例2
输入:n = 1
输出:True
示例3
输入:n = 2
输出:True
关键思路
让我们来逆推一下,玩家的胜利条件是什么:
玩家需要在自己的倒数第二个回合后,将石头的数量控制为4,这样无论朋友拿走1块,2块还是3块石头,最后一块石头都将被玩家拿走。
那么,玩家需要在自己的倒数第三个回合后,将石头的数量控制为8,这样无论朋友拿走1块,2块还是3块石头,玩家都可以做到在自己的倒数第二个回合后,将石头的数量控制为4。
以此类推,我们知道玩家的胜利条件是,在自己的每个回合后,将石头的数量控制在 4的倍数 。由于玩家是先手,每个回合可以拿走1块,2块或是3块石头,所以当给定石头数量为 4的整倍 时,玩家无法在回合后将石头的数量控制在 4的倍数 ;同时,朋友在每个回合也将采取同玩家相同的最优策略,在这种情况下,玩家将输掉游戏。
代码实现
class Solution(object):
def canWinNim(self, n):
"""
:type n: int
:rtype: bool
"""
return False if n%4==0 else True
运行结果
False
True
True
总结反思
第一眼看到题目的时候在想是否需要用到递归或者动态规划的思路,但后来发现只需判断玩家是否有胜算。于是我立刻想到了之前在经济课中学到的“海盗分赃”,运用相同的思路将题解了出来。