Leetcode 464.我能赢吗

时间:2023-09-06 21:10:38

我能赢吗

在 "100 game" 这个游戏中,两名玩家轮流选择从 1 到 10 的任意整数,累计整数和,先使得累计整数和达到 100 的玩家,即为胜者。

如果我们将游戏规则改为 "玩家不能重复使用整数" 呢?

例如,两个玩家可以轮流从公共整数池中抽取从 1 到 15 的整数(不放回),直到累计整数和 >= 100。

给定一个整数 maxChoosableInteger (整数池中可选择的最大数)和另一个整数 desiredTotal(累计和),判断先出手的玩家是否能稳赢(假设两位玩家游戏时都表现最佳)?

你可以假设 maxChoosableInteger 不会大于 20, desiredTotal 不会大于 300。

示例:

输入:

maxChoosableInteger = 10

desiredTotal = 11

输出:

false

解释:

无论第一个玩家选择哪个整数,他都会失败。

第一个玩家可以选择从 1 到 10 的整数。

如果第一个玩家选择 1,那么第二个玩家只能选择从 2 到 10 的整数。

第二个玩家可以通过选择整数 10(那么累积和为 11 >= desiredTotal),从而取得胜利.

同样地,第一个玩家选择任意其他整数,第二个玩家都会赢。

DFS + memorizatoin

通过state保存状态,即当前选了哪些数

需要注意maxChoosableInteger不会大于20,我们完全可以通过一个32位的整数来表示状态

例如选择了1和2,那么状态为01 | 10 = 11 = 3(选择的数为1左移位数-1)

 class Solution {
public boolean canIWin(int maxChoosableInteger, int desiredTotal) {
if(desiredTotal <= maxChoosableInteger)
return true;
//Note: n should be <= 32 as int is 32-bit in Java; else it will 1 << 33+ equals 0.
int n = maxChoosableInteger;
int sum = n * (n + 1) / 2;
if(sum < desiredTotal)
return false;
Boolean[] dp = new Boolean[1 << n];
return canIWin(0, n, desiredTotal, dp);
} private boolean canIWin(int state, int n, int remain, Boolean[] dp) {
if (remain <= 0) {
//dp[state] = false;
// Base case:
return false;
}
if (dp[state] == null) {
dp[state] = false;
int mask = 1;
//Key Point: take from the tail
for(int i = 1; i <= n; i++){
int future = state | mask;
//the other can win
if (future != state && !canIWin(future, n, remain - i, dp)) {
//update current status = true
dp[state] = true;
break;
}
mask <<= 1;
}
}
return dp[state];
}
}