问题描述
你和你的朋友,两个人一起玩 Nim 游戏:桌子上有一堆石头,每次你们轮流拿掉 1 - 3 块石头。 拿掉最后一块石头的人就是获胜者。你作为先手。
你们是聪明人,每一步都是最优解。 编写一个函数,来判断你是否可以在给定石头数量的情况下赢得游戏。
示例:
输入: 4
输出: false
解释: 如果堆中有 4 块石头,那么你永远不会赢得比赛;
因为无论你拿走 1 块、2 块 还是 3 块石头,最后一块石头总是会被你的朋友拿走。
问题分析
首先,题目默认是我先手拿石头,已知一二三块石头的时候我可以一次拿完,直接获得胜利。然后当有四块石头的时候,显然我无论怎么拿,对手都可以把剩下的直接拿完。有意思的是,我们已知一二三必赢,四必输的情况下,如果是五块的情况下,我们可以只拿一块把必输的情况留给对手,六七同理。然而当八时,无论我们怎么拿,都无法避免对手把必输的情况留给我们。同理以此类推可得四的倍数的情况下必输,其他胜利
代码
class Solution:
def canWinNim(self, n: int) -> bool:
#用与运算加快运算速度
return n&3