LeetCode 377. Combination Sum IV
Solution1:
我的未能AC的答案
题目描述的和前几道题差不多,但实际上不能用DFS来做(会超时),要用动态规划,还是记录一下吧。。。
class Solution {
public:
int combinationSum4(vector<int>& nums, int target) {
int num = 0;
vector<int> out;
combinationSumDFS(nums, target, out, num);
return num;
}
void combinationSumDFS(vector<int> &candidates, int target,
vector<int> &out, int& num) {
if (target < 0) return;
else if (target == 0) num++;
else {
for (int i = 0; i < candidates.size(); ++i) {
out.push_back(candidates[i]);
combinationSumDFS(candidates, target - candidates[i], out, num);
out.pop_back();
}
}
}
};
Solution2:
参考网址:http://www.cnblogs.com/grandyang/p/5705750.html
这道题是组合之和系列的第四道,我开始想当然的一位还是用递归来解,结果写出来发现TLE了。
这道题的真正解法应该是用DP来做,解题思想有点像之前爬梯子的那道题Climbing Stairs,我们需要一个一维数组dp,其中dp[i]表示目标数为i的解的个数,然后我们从1遍历到target,对于每一个数i,遍历nums数组,如果i>=x, dp[i] += dp[i - x]。这个也很好理解:
比如说对于[1,2,3] 4,这个例子,当我们在计算dp[3]的时候,3可以拆分为1+x,而x即为dp[2],3也可以拆分为2+x,此时x为dp[1],3同样可以拆为3+x,此时x为dp[0],我们把所有的情况加起来就是组成3的所有情况了,参见代码如下:
class Solution {
public:
int combinationSum4(vector<int>& nums, int target) {
vector<int> dp(target + 1);
dp[0] = 1;
for (int i = 1; i <= target; ++i) {
for (auto a : nums) {
if (i >= a) dp[i] += dp[i - a];
}
}
return dp.back();
}
};