Leetcode 464. Can I Win

来源:互联网 发布:淘宝最过的零食店 编辑:程序博客网 时间:2024/06/01 10:08

In the “100 game,” two players take turns adding, to a running total, any integer from 1..10. The player who first causes the running total to reach or exceed 100 wins.

What if we change the game so that players cannot re-use integers?

For example, two players might take turns drawing from a common pool of numbers of 1..15 without replacement until they reach a total >= 100.

Given an integer maxChoosableInteger and another integer desiredTotal, determine if the first player to move can force a win, assuming both players play optimally.

You can always assume that maxChoosableInteger will not be larger than 20 and desiredTotal will not be larger than 300.

Example

Input:
maxChoosableInteger = 10
desiredTotal = 11

Output:
false

Explanation:
No matter which integer the first player choose, the first player will lose.
The first player can choose an integer from 1 up to 10.
If the first player choose 1, the second player can only choose integers from 2 up to 10.
The second player will win by choosing 10 and get a total = 11, which is >= desiredTotal.
Same with other integers chosen by the first player, the second player will always win.
原文链接
思路分析:
状态压缩的动态规划算法 同时也是 带记忆化的dfs算法。数字选取范围最大为不超过20,我们可以用整数的每一位代表一个数字是否被选择,这样就可以用一个整数代表所有数被选择的状态。注意到如果用数组表示,范围太大,所以我们用map表示。
首先判断两种边界情况:第一次选择就成功 和 不可能成功

代码如下:

class Solution {private:    int maxn;    map<int,bool> mp;public:    bool canIWin(int maxChoosableInteger, int desiredTotal) {        maxn = maxChoosableInteger;        if(maxn >= desiredTotal) return true;        if((1 + maxn) * maxn / 2 < desiredTotal) return false;        return canWin(desiredTotal,0);    }    bool canWin(int target, int visited) {        if(mp.find(visited) != mp.end()) return mp[visited];        for(int i = 1; i <= maxn; i++) {            int mask = (1 << (i-1));            if((mask & visited) == 0 && (i >= target || canWin(target-i, mask | visited) == false)) {                mp[visited] = true;                return true;            }        }        mp[visited] = false;        return false;    }};
原创粉丝点击