leetcode 377 组合总和IV

leetcode 377 组合总和IV

题目：给你一个由 不同 整数组成的数组 nums ，和一个目标整数 target 。请你从 nums 中找出并返回总和为 target 的元素组合的个数。

示例 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

思路：

这个题是一个很明显的动态规划题目。我的初步思路是：

• 构造一个大小为target的数组dp，对于小于target且在数组中的数字i，初始化dp[i]=1，其他为0;

• 数字A，一定是由小于他的数字i，和数字A-i构成的，那么数字A的组合情况总数就是

for(i=1;i<A/2;i++){
    if (dp[i]!=0&&dp[A-i]!=0) dp[A]=dp[i]+dp[A-i];
}

• 因此我们可以自低向上进行计计算

但是实际上这个思路是有纰漏的：

• 数组为[1,3]
• 初始化时候dp[1]=1,dp[3]=1
• 明显根据上述计算规则dp[1]=1;dp[2]=1;
• dp[3]=dp[3]+dp[1]+dp[2]=3

但是实际上dp[3]组合为两种：

3
1 1 1

• 这是因为上述思路中dp[2]为1，在计算dp[3]的时候，误以为2也在数组中。
• 所以对于原本dp[j]=0的数就不计算其dp是之还是为0。
• 但这样的话在上例中dp[2]就为0，dp[3]最后为1。还是不正确

这时候再换一个思路，参考如下题解思路

class Solution {
public:
    int combinationSum4(vector<int>& nums, int target) {
        vector<int> dp(target + 1);
        dp[0] = 1;
        for (int i = 1; i <= target; i++) {
            for (int& num : nums) {
                if (num <= i && dp[i - num] < INT_MAX - dp[i]) {
                    dp[i] += dp[i - num];
                }
            }
        }
        return dp[target];
    }
};