题目:组合总和 Ⅳ
- 给定一个由正整数组成且不存在重复数字的数组,找出和为给定目标正整数的组合的个数。
示例:
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。
进阶:
- 如果给定的数组中含有负数会怎么样?
- 问题会产生什么变化?
- 我们需要在题目中添加什么限制来允许负数的出现?
来源:力扣(LeetCode)第377题
分析:
- dp和dfs两种方法。
代码:
dp
class Solution { public int combinationSum4(int[] nums, int target) { int len = nums.length; int[] dp = new int[target+1]; dp[0] = 1; for (int i = 1; i <= target; i++) { for (int num : nums) if (i >= num) dp[i] += dp[i-num]; } return dp[target]; } }
dfs + 记忆化
另外强调这种方法用java会超时,用c++要注意不要把数组初始化为0,就用数组的原数。
class Solution { public: int len; int* memo; int combinationSum4(vector<int>& nums, int target) { len = nums.size(); memo = new int[target]; return dfs(nums, 0, target); } int dfs(vector<int>& nums, int sum, int target) { if (sum == target) return 1; if (sum > target) return 0; if (memo[sum] != -1094795586) return memo[sum]; int ans = 0; for (int i = 0; i < len; i++) ans += dfs(nums, sum + nums[i], target); memo[sum] = ans; return ans; } };