【LeetCode】377. 组合总和 Ⅳ

题目描述

给定一个由正整数组成且不存在重复数字的数组，找出和为给定目标正整数的组合的个数。

示例：

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。

 

思路

本题用递归是会超时的，只能用动态规划，这题同 279. 完全平方数 类似，dp表示的是当前target的组合数，也就是所求。

dp[0]=1是因为只要减到0了就证明这是一个组合，所以要加1，所有dp初始化为0是因为要叠加所以起始是0。

因为nums中每个元素不同，所以对于每个target的递归方程为 dp[ i ] += dp[ i - nums[ j ] ]，对每个target遍历所有元素，其实相当于看看当前target减去每个元素之后的组合数，再把他们加起来，就是当前target的组合数，这也是动态规划的原理。又因为允许重复，所以再遍历过程中只要求个数就行，重复不重复不要紧。

比如nums为[1,2]，target为3，易知dp[1]=1，表示组合[1]，dp[2]=2，表示组合[1,1]和[2]，那么求dp[3]的时候其实就是在dp[1]的所有组合上加了2，在dp[2]的所有组合上加了1，也就得到了3个组合[1,2]、[1,1,1]、[2,1]，可见有重复的，但是题目里说没影响，所以这种方法可行。

 

代码

class Solution {
public:
    int combinationSum4(vector<int>& nums, int target) {
        int n = nums.size();
        vector<unsigned long long> dp(target+1,0);
        dp[0]=1;
        for(int i=1;i<=target;i++){
            for(int j=0;j<n;j++){
                if(nums[j]<=i){
                    dp[i] += dp[i-nums[j]];
                }
            }
        }
        return dp[target];
    }
};