本文记录的是刷题过程中的重要概念和笔记。如有侵权,请联系删除。
377. 组合总和 Ⅳ
力扣题目链接(opens new window)
难度:中等
给定一个由正整数组成且不存在重复数字的数组,找出和为给定目标正整数的组合的个数。
示例:
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。
思路
其实就是求排列!
组合不强调顺序,(1,5)和(5,1)是同一个组合。
排列强调顺序,(1,5)和(5,1)是两个不同的排列。
如果本题要把排列都列出来的话,只能使用回溯算法爆搜。
DP五部曲
1.确定dp数组(dp table)以及下标的含义
dp[i]: 凑成目标正整数为i的排列个数为dp[i]
2.递推公式
dp[i] += dp[i - nums[j]];
3.dp数组初始化
因为题目中也说了:给定目标值是正整数! 所以dp[0] = 1是没有意义的,仅仅是为了推导递推公式。
至于非0下标的dp[i]应该初始为多少呢?
初始化为0,这样才不会影响dp[i]累加所有的dp[i - nums[j]]。
4.遍历顺序
如果求组合数就是外层for循环遍历物品,内层for遍历背包。
如果求排列数就是外层for遍历背包,内层for循环遍历物品。
5.举例推导
dp【4】 = dp[3] (在dp【3】的排列后加上1构成新的排列) + …
区分:求组合数(外层for循环遍历物品,内层for遍历背包)
在原有数字的基础上再多使用数字nums[i]
C++
class Solution {
public:
int combinationSum4(vector<int>& nums, int target) {
vector<int> dp(target + 1, 0);
dp[0] = 1;
for (int i = 0; i <= target; i++) { // 遍历背包
for (int j = 0; j < nums.size(); j++) { // 遍历物品
if (i - nums[j] >= 0 && dp[i] < INT_MAX - dp[i - nums[j]]) {
dp[i] += dp[i - nums[j]];
}
}
}
return dp[target];
}
};
C++测试用例有两个数相加超过int的数据,所以需要在if里加上dp[i] < INT_MAX - dp[i - num]。