经典博弈

来源:互联网 发布:java date format 编辑:程序博客网 时间:2024/06/05 18:24

描述
有两个人玩游戏,给定一个最大可取代数maxChoosableInteger,两个人轮流从1~maxChoosableInteger中取一个数,取过的数不可再取,若其中一方取过以后,所有取过的数的和大于等于desiredTotal,那么这个人获胜。现在给maxChoosableInteger和desiredTotal,问先手是否必胜,假定游戏双方均采取最优策略。你可以假定给出的maxChoosableInteger不超过20,desiredTotal不超过300。


样例
输入:
maxChoosableInteger = 10
desiredTotal = 11
输出:
false


说明
无论先手怎么取数,先手都会输掉游戏。先手可以取到1到10中的任何一个。如果先手取1,那么后手可以取2到10中任何一个。后手如果取10,那么就可以赢得游戏。假如先手取其它的数,那么后手仍然能赢得游戏。


思路
明显的是,这个全排列问题不能用枚举法来做。1~n的前缀后就是1~n-1的和加上n,用记忆化搜索判断胜利情况。要么当前剩下的desiredTotal小于等于0,要么对于剩下的还未取得数,已经搜索过且是必胜的状态。假设这个状态没有搜索过,就进行判断这个状态,直到遇到判断过的状态或desiredTotal小于等于0。


代码

#include "stdafx.h"#include<iostream>#include<vector>#include<algorithm>using namespace std;class Solution {public:    vector<int> dp;    vector<bool> used;    bool boyi(int maxChoosableInteger, int desiredTotal) {        int sum = (1 + maxChoosableInteger)*maxChoosableInteger / 2;        if (sum < desiredTotal) {            return false;        }        if (desiredTotal <= maxChoosableInteger) {            return true;        }        dp.resize(1 << maxChoosableInteger);        fill(dp.begin(),dp.end(),-1);        used.resize(maxChoosableInteger + 1);        fill(used.begin(), used.end(), 0);        return handler(desiredTotal);    }    bool handler(int desiredTotal) {        if (desiredTotal <= 0)            return false;        int key = fmt(used);        if (dp[key] == -1) {            for (int i = 1; i < used.size(); i++) {                if (!used[i]) {                    used[i] = true;                    if (!handler(desiredTotal - i)) {                        dp[key] = -1;                        used[i] = false;                        return true;                    }                    used[i] = false;                }            }            dp[key] = 0;        }        return dp[key] == 1;    }    int fmt(vector<bool> & used) {        int num = 0;        for (int i = 1; i < used.size(); i++) {            num <<= 1;            if (used[i]) {                num |= 1;            }        }        return num;    }};int main(){    int maxChoosableInteger, desiredTotal;    cin >> maxChoosableInteger >> desiredTotal;    Solution sol;    bool result=sol.boyi(maxChoosableInteger, desiredTotal);    if (result) {        cout << "true" << endl;    }    else {        cout << "false" << endl;    }    return 0;}