dfs超时,这不是普通的完全背包问题。
还是考虑用记忆化搜索比较靠谱!!如果接下来要搜索的数字组合数搜索过了,就直接返回。
记录mem[target]就是用一个res记录从当前的target dfs下去的所有可行的结果之和,把res赋值给mem[target],递归边界有两个,当target<0就返回0。或者mem[target]!=-1就返回,所以我们一开始就把mem[0]给赋值为1,这样当target为0时,证明有一种可行解,直接返回1了。
class Solution {
public:
int dfs(vector<int>& nums, int target){
//这也就是递归边界
if(target < 0) return 0;
//这也是一个递归边界,当target==0时,返回1,代表多了一种结果
if(mem[target] != -1) return mem[target];
int res = 0;
for(int i = 0; i < nums.size(); ++i){
res += dfs(nums,target-nums[i]);
}
mem[target] = res;
return res;
}
int combinationSum4(vector<int>& nums, int target) {
//dfs,每次继续从第0个开始即可
//用记忆化,比如(1,2,3),32,那么32-两个1之后,我们要求dfs(30),32-2也是30,所以会重复计算,开一个备忘录
//一开始赋为-1,basecase是mem[0] = 1,找总额为0时一种方法
mem.resize(target+1,-1);
mem[0] = 1;
return dfs(nums,target);
}
private:
vector<int> mem;
};