题目:组合总和Ⅳ
描述:
示例 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,1,1,1),(1,1,2),(1,3),(2,2)。因为题目说(1,3)和(3,1)是两个不同的答案。这道题的难点就是在这。
因此我就想,有没有什么办法能够忽略掉这个排列组合问题?于是我想到了背包问题。背包问题中,它在本次取背包的过程中,并不要求你之前是如何取背包的。因此我们可以用一个数组dp,dp[i]表示组合和为i的个数。对于任意一个dp[i],我们遍历nums,如果nums[j]<=i,我们就可以将dp[i-nums[j]]的值加到dp[i]中。特别地,dp[0] = 1。
int dp[target+1];
memset(dp,0,sizeof(dp));
dp[0]=1;
for(int i = 1 ; i <= target ;i++){
for(int j = 0 ; j < numsSize ;j++){
if(nums[j]<=i&&dp[i-nums[j]]<INT_MAX-dp[i]){
dp[i]+=dp[i-nums[j]];
}
}
}
return dp[target];
这里的i-nums[j]就代表以nums[j]为结尾的和为i的组合。
可能有小伙伴会有疑问,这真的包含了所有组合吗?比如上面提到的(1,3)和(3,1)。答案是正确的。因为我们i-nums[j]表示的是以nums[j]结尾的组合。在内层的循环中,我们遍历了整个数组,1和3都有可能作为该组合的结尾。所以包含了所有可能性。
题目中还有一个关键点,如果没有这一句的话,答案会溢出。
dp[i-nums[j]]<INT_MAX-dp[i]