DP的两种解法,一种是递归记忆型,另一种是递推型,递推型又分为人人为我和我为人人型。
题目简述
给定一个两个数,分别是maxChoosableInteger、desiredTotal,有两个人,依次从1-maxChoosableInteger中取数,不放回,最后使得取出的数的和大于或等于desiredTotal的那个人是赢家,问是否第一个取数的人能赢。PS:每个人都是取数高手。
题目理解
通过尝试用其他更直观的语言描述问题,可以帮助我们找到问题的解法。假设我是第一个人,我取1,如果第二个人取剩下的某个数能够保证他稳赢,即当我在取下一个数时,不管这个数是什么,我都是输,那么我不能取1。如何确定第二个人取剩下的某个数能够保证他稳赢呢?那么我们就得看当第二个人取了这个数后我再取其他任意数是不是都是输,如果是,那么第二个人取这个数稳赢。是不是感觉到有递归的层次感,那么我们再思考递归的终止条件是啥,因为我们需要遍历所有的情况,所以递归的终止条件就是简单的当和大于等于desiredTotal时。
问题分析
对于这个问题,没有什么取数技巧,不要想当然地以为存在什么取数技巧只谈,就不敢动手动脑了。其实就是简单的DFS,讨论每一种情况,使用递归,但存粹的递归效率低,且存在重复计算的问题,例如第一个人取1、第二个人取2与第一个人取2、第二个人取1这两种情况是重复的,题目说了maxChoosableInteger最大为20,递归深度不会太深,所以我们可以采用记忆式的递归方法。
题目的技巧
为了讨论每一种情况,我们需要记录哪些元素使用了,可以使用一个20大小的数组充当hash表存储。比较快而简单的做法是使用一个整型的位去存储,1表示使用了,0表示未使用,且32位大于20。
代码
class Solution {
public:
vector<char>* v;
bool canIWin(int maxChoosableInteger, int desiredTotal) {
v = new vector<char>(1 << maxChoosableInteger, -1);
if((maxChoosableInteger+1)*maxChoosableInteger/2 < desiredTotal)
return false;
return rc(maxChoosableInteger, desiredTotal, 0, 0);
}
bool rc(int maxChoosableInteger, int desiredTotal, int used, int curSum)
{
//if generate error,you may not consider some edge testcase
if(v->at(used) != -1)
return v->at(used);
for(int i = 0; i < maxChoosableInteger; i++)
{
if(!((1 << i) & used))
{
if(desiredTotal <= curSum + i + 1 || !rc(maxChoosableInteger,
desiredTotal,
used|(1<<i),
curSum+i+1))
{
(*v)[used] = 1;
return true;
}
}
}
(*v)[used] = 0;
return false;
}
};
总结
- 不要过分相信题目有一种不为人知的技巧而不敢动手,一般题目也不会包含一些鲜为人知的技巧,要敢于思考其他方法,敢于动手;
- 对于一些基础算法例如DFS算法,要熟练掌握。
- 如果程序出现错误,也可能是自己的一些边界值没有考虑到。
- 可以使用一个int类型充当只有两种取值的hash表。
- 位运算对于提高程序的效率很重要。
- 400万字节才3mb多,对于这种题目作为dp数组还是不错的。