题目链接:https://leetcode-cn.com/problems/combination-sum-iv/
题目描述
给定一个由正整数组成且不存在重复数字的数组,找出和为给定目标正整数的组合的个数。
示例:
nums = [1, 2, 3]
target = 4
所有可能的组合为:
(1, 1, 1, 1)
(1, 1, 2)
(1, 2, 1)
(1, 3)
(2, 1, 1)
(2, 2)
(3, 1)
请注意,顺序不同的序列被视作不同的组合。
因此输出为 7。
思路
动态规划
这个题只要返回总数即可,不用返回具体的组合;这是我们考虑记忆化搜索,用动态规划来解决。
用一维数组dp[n]
表示target
为n
时的组合总数
动态规划方程:
dp[n] += dp[n - num] for num in nums
注意:在 0
这一点,我们是定义 dp[0] = 1
的,因为它表示如果 nums
里有一个数恰好等于 target
,它单独成为 1
种可能。
代码
class Solution {
public:
int combinationSum4(vector<int>& nums, int target) {
if(nums.empty()) return 0;
vector<long > dp(target+1,0);
dp[0] = 1;
for (int i = 1; i <= target; ++i) {
for (int j = 0; j < nums.size(); ++j) {
if(i >= nums[j])
dp[i] += dp[i-nums[j]] %INT_MAX;
}
}
return dp[target];
}
};