令maxn=maxChooseableInteger,total=desiredTotal。
如果maxn大于等于total,那么先手一次拿取就获胜,直接返回。
如果1+2...maxn都是小于total的 ,那么无论怎么取,先手都不能获胜,直接返回。
对于博弈论的函数,一般模式我们判定需要的是:
check(...当前状态)= true if not check(下一状态,也就是对手的状态)else false
为最基本的模板,这里因为要时刻记住1~maxn的数字到底被用过了没有,并且要记录当前状态下是赢了还是输了,因此我们要用一个位图来记录,用一个2**n长度数组记录结果。
class Solution:
def dfs(self,status,dp,total,num):
#如果当前状态的 胜负已经被计算过了,直接返回即可
if dp[status]!=None:
return dp[status]
for i in range(1,num+1):
b=1<<(i-1)#取第i个数字,如果已经被用过了直接忽略,status的位记录了已经被使用的
if status& b:
continue
#如果i能够使用并且已经大于等于total了,那么已经是赢了的
#否则需要下一状态要对手输了才行
if i>=total or not self.dfs(status|b,dp,total-i,num):
dp[status]=True
return True
dp[status]=False
return False
def canIWin(self, num: int, total: int) -> bool:
#两种特殊情况
if num>=total:
return True
if (1+num)*num//2 < total:
return False
return self.dfs(0,[None]*(1<<num),total,num)