Leetcode 464.我能赢吗

我能赢吗

在 "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];
    }
}