leetcode 377.组合综合IV - 动态规划 - 完全背包
题干
给你一个由 不同 整数组成的数组 nums ,和一个目标整数 target 。请你从 nums 中找出并返回总和为 target 的元素组合的个数。
题目数据保证答案符合 32 位整数范围。
示例 1:
输入:nums = [1,2,3], target = 4
输出:7
解释:
所有可能的组合为:
(1, 1, 1, 1)
(1, 1, 2)
(1, 2, 1)
(1, 3)
(2, 1, 1)
(2, 2)
(3, 1)
请注意,顺序不同的序列被视作不同的组合。
示例 2:
输入:nums = [9], target = 3
输出:0
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 1000
nums 中的所有元素 互不相同
1 <= target <= 1000
进阶:如果给定的数组中含有负数会发生什么?问题会产生何种变化?如果允许负数出现,需要向题目中添加哪些限制条件?
知识点&算法
看到集合数很容易想到dfs,但是这题数据量过大了,而且还是数可以重复选的情况。
dfs不同立马往dp想,如果用dp来维护前缀组合的个数,同时还要标记前缀组合的值,比较难实现,或者说复杂度还是有点高。
那么像能不能用dp[i]来维护和为i时的组合个数,看起来是可行的。
考虑边界条件:
显然dp[0]可以什么都不选,置1,为了统计个数,其他全部置0
考虑答案:
显然是dp[target]
考虑状态转移方程:
遍历组合的和i,由于数组中数可以任意次出现,所以每遍历一个i都要遍历所有的num(类似完全背包的思想,也可以先对数组进行排序预处理,这样在内层遍历的时候可以通过下标来控制num的大小合法)
对于遍历到的num来说(当i - num >= 0 时),dp[i] = dp[i] + dp[i-num]; 意味着num可以接在所有和为i-num的组合的最后,即在和为i-num的组合的最后添了一位。
这里有个很离谱的坑:题目保证了答案在32位整数内,但是某些样例的中间数据可能超出了32位整数的范围(甚至超出64位整数),所以为了通过这题,需要将溢出的数据忽略不计,即在合法判断里加上这一条件。挺没意思的。
题解
class Solution {
public:
int combinationSum4(vector<int>& nums, int target) {
int n = nums.size(),res = 0;
vector<int> dp(target+5,0);
dp[0] = 1;
for(int i = 1 ; i <= target ; ++i){
for(auto num : nums){
if(i - num >= 0 && dp[i] < INT_MAX - dp[i-num]){
dp[i] += dp[i-num];
}
}
}
return dp[target];
}
};