LeetCode 464. Can I Win

题目

题目地址

题目给定了一个游戏，规则是两个人玩这个游戏，玩之前先定好两个数字maxChoosableInteger和desiredTotal，玩家可以从1～maxChoosableInteger中任选一个数字，任何人选过一个数字后，这个数字就不能再被选中。两个人轮流选中，直到总和超过desiredTotal，最后一个选的人赢。问题是，给出maxChoosableInteger和desiredTotal，问第一个选的人是否能赢得比赛。

例子是输入10,11，那么无论第一个选的人如何选择，他都不可能赢，所以输出false。

分析

首先有个特殊情况是所有可选数字加起来也达不到desiredTotal，这种情况输出false，因为谁都不可能赢。除此之外，刚刚看到题目时，我一直没有想到好的思路，参考网上的分析我才明白，本题的关键是无论游戏中的哪一方，在面临相同的情况时（已被选的数字相同），最后结果都一样。既然如此就可以分解问题，玩家在选择时，一一选中所有未被选中的数字，如果赢了，或者，如果这样选择对方会输，那么返回赢，结束；否则，返回输。这个逻辑很好理解，关键在于判断对方赢否其实是可以在统一的框架内解决，使用递归解决，本题有意思的地方是看似好像要用贪心，或者有什么数学方法，其实用记忆化搜索即可。

实现

#include <iostream>
#include <vector>
#include <map>

using namespace std;

class Solution {
    vector<map<int, bool> > dp;
    int maxInt;

    bool search(int desired, int status) {
        if (dp[desired].count(status) != 0)
            return dp[desired][status];

        int bit;
        for (int i = maxInt; i >= 1; --i) {
            bit = 1 << i;
            if (status & bit) {
                if (i >= desired || !search(desired - i, status ^ bit))
                    return dp[desired][status] = true;
            }
        }

        return  dp[desired][status] = false;
    }

  public:
    bool canIWin(int maxChoosableInteger, int desiredTotal) {
        maxInt = maxChoosableInteger;

        int maxStatus = 0;
        for (int i = 1; i <= maxInt; ++i)
            maxStatus |= 1 << i;

        if (maxInt * (maxInt + 1) / 2 < desiredTotal)
            return false;

        dp.assign(desiredTotal + 1, map<int, bool>());

        return search(desiredTotal, maxStatus);
    }
};

在初始的实现中，因为search有两个形参，所以很直接地开了二维的记忆数组，当然用了map，因为如果没有搜索到就不会添加到map中，这样就能简单地判断是否已被搜索，通过的时间是99ms。看了一些其他的分析，发现确实可以不用第一个参数，因为这两个参数是关联的，而且，因为desiredTotal相同，可以很容易地根据status参数计算出唯一的desired，开记忆数组时可以省去，于是我改成了单独一个map

#include <iostream>
#include <vector>

using namespace std;

class Solution {
    vector<bool> dp;
    vector<bool> visited;
    int maxInt;

    bool search(int desired, int status) {
        if (visited[status])
            return dp[status];

        int bit;
        for (int i = maxInt; i >= 1; --i) {
            bit = 1 << i;
            if (status & bit) {
                if (i >= desired || !search(desired - i, status ^ bit)) {
                    visited[status] = true, dp[status] = true;
                    return true;
                }
            }
        }

        visited[status] = true, dp[status] = false;
        return false;
    }

  public:
    bool canIWin(int maxChoosableInteger, int desiredTotal) {
        maxInt = maxChoosableInteger;

        if (maxInt * (maxInt + 1) / 2 < desiredTotal)
            return false;

        int maxStatus = (1 << (maxInt + 1)) - 2;
        dp.assign(maxStatus + 1, false);
        visited.assign(maxStatus + 1, false);

        return search(desiredTotal, maxStatus);
    }
};