在 "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),从而取得胜利.
同样地,第一个玩家选择任意其他整数,第二个玩家都会赢。
思路:使用哈希表来记录已经计算过的结果。
1、如果给定的数字范围大于等于目标值的话,直接返回 true。
2、如果给定的数字总和小于目标值的话,说明谁也没法赢,返回 false。
3、然后我们进入递归函数,首先我们查找当前情况是否在 HashMap 中存在,有的话直接返回结果。
没有的话就需要求了:
由于maxChoosableInteger
不会大于 20,我们使用一个字符串used来记录数组中的某个数字是否使用过。遍历所有没有使用过的数字,如果此时的目标值小于等于当前数字,或者调用递归函数,如果返回 false,说明已经赢了。如果遍历完所有数字,都没找到,则标记 false,并返回 false。
class Solution {
public:
bool canWin(int length, int total, unordered_map<string, bool> &m, string &used) {
if (m.count(used)) return m[used];
for (int i = 0; i < length; ++ i) {
if (used[i] == '0') {
used[i] = '1';
if (i + 1 >= total || !canWin(length, total-i-1, m, used)) {
used[i] = '0';
m[used] = true;
return true;
}
used[i] = '0';
}
}
m[used] = false;
return false;
}
bool canIWin(int maxChoosableInteger, int desiredTotal) {
if (maxChoosableInteger >= desiredTotal) return true;
if ((1 + maxChoosableInteger) * maxChoosableInteger / 2 < desiredTotal)
return false;
unordered_map<string, bool> m;
string used = string(maxChoosableInteger, '0');
return canWin(maxChoosableInteger, desiredTotal, m, used);
}
};