leetcode464. 我能赢吗
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/can-i-win
题目描述
在 “100 game” 这个游戏中,两名玩家轮流选择从 1 到 10 的任意整数,累计整数和,先使得累计整数和 达到或超过 100 的玩家,即为胜者。
如果我们将游戏规则改为 “玩家 不能 重复使用整数” 呢?
例如,两个玩家可以轮流从公共整数池中抽取从 1 到 15 的整数(不放回),直到累计整数和 >= 100。
给定两个整数 maxChoosableInteger (整数池中可选择的最大数)和 desiredTotal(累计和),若先出手的玩家能稳赢则返回 true ,否则返回 false 。假设两位玩家游戏时都表现 最佳 。
示例 1:
输入:maxChoosableInteger = 10, desiredTotal = 11
输出:false
解释:
无论第一个玩家选择哪个整数,他都会失败。
第一个玩家可以选择从 1 到 10 的整数。
如果第一个玩家选择 1,那么第二个玩家只能选择从 2 到 10 的整数。
第二个玩家可以通过选择整数 10(那么累积和为 11 >= desiredTotal),从而取得胜利.
同样地,第一个玩家选择任意其他整数,第二个玩家都会赢。
示例 2:
输入:maxChoosableInteger = 10, desiredTotal = 0
输出:true
示例 3:
输入:maxChoosableInteger = 10, desiredTotal = 1
输出:true
提示:
1 <= maxChoosableInteger <= 20
0 <= desiredTotal <= 300
解题思路
考虑边界情况,当所有数字选完仍无法到达 desiredTotal 时,两人都无法获胜,返回 false。当desiredTotal 小于等于 maxChoosableInteger 时,先手方直接获胜。当所有数字的和大于等于 desiredTotal时,其中一方能获得胜利,需要通过搜索来判断获胜方。
在游戏中途,假设已经被使用的数字的集合为 usedNumbers,这些数字的和为 currentTotal。当某方行动时,如果他能在未选择的数字中选出一个 i,使得 i+currentTotal≥desiredTotal,则他能获胜。否则,需要继续通过搜索来判断获胜方。在剩下的数字中,如果他能选择一个 i,使得对方在接下来的局面中无法获胜,则他会获胜。否则,他会失败。
根据这个思想设计搜索函数 process,其中 usedNumbers 可以用一个整数来表示,从低位到高位,第 i 位为 1 则表示数字 i 已经被使用,为 0 则表示数字 i 未被使用。如果当前玩家获胜,则返回 true,否则返回 false。为了避免重复计算,需要使用记忆化的操作来降低时间复杂度。
代码演示
class Solution {
//记忆化的hashMap,用来记录选择了哪些树的情况,当前玩家的胜负;用以减少时间复杂度
Map<Integer, Boolean> memory = new HashMap<>();
//主方法
public boolean canIWin(int maxChoosableInteger, int desiredTotal) {
//目标数字小于等于当前最大数字,先手肯定赢。
if(desiredTotal <= maxChoosableInteger){
return true;
}
//如果加起来都不能超过的话,都不可能赢,return false
if ((1+maxChoosableInteger)*maxChoosableInteger*0.5<desiredTotal){
return false;
}
//否则的话递归判断,判断先手能不能赢,需要注意的是,先手从0开始
return process(maxChoosableInteger,0,desiredTotal,0);
}
/**
* @param maxChoosableInteger 供选择的数
* @param userNumbers 被选择的数,注意用二进制表示,可能比较难以理解。如0010则表示2已经被选择,0001则表示1已被选择
* @param desiredTotal 累计整数合需要超过的目标值
* @param currentTotals 当前累计整数合
*/
public boolean process(int maxChoosableInteger, int userNumbers, int desiredTotal, int currentTotals){
if (!memory.containsKey(userNumbers)){
boolean ans = false;
//随机选择其中一个数字,所以要循环下 从0开始循环,要选择的数字应该是 i + 1
for (int i = 0; i < maxChoosableInteger;i++){
//判断这个数字是否被选择了
if (((userNumbers >> i) & 1) == 0){
//判断选择后是否可以直接获胜
if(i + 1 + currentTotals >= desiredTotal){
ans = true;
break;
}
//将选择的数字 加入到已经选择的数字中
//这里用二进制位上 有没有1来确定是否被选 userNumbers | (1 << i)
//已经选择的数字总和 currentTotals = 1 + i + currentTotals;
//递归让对手去选择
// boolean process = process(maxChoosableInteger, userNumbers, desiredTotal, currentTotals);
//对手输了的话,自己就赢了,所以取反
if(!process(maxChoosableInteger, userNumbers | (1 << i),
desiredTotal, 1 + i + currentTotals)){
ans = true;
break;
}
}
}
memory.put(userNumbers,ans);
}
return memory.get(userNumbers);
}
}