Skip to content

377. 组合总和 Ⅳ Medium

给你一个由 不同 整数组成的数组 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

解题思路

输入: 一个不同整数数组 nums,一个目标整数 target

输出: 返回总和为 target 的元素组合个数

本题属于线性DP问题。

1. 将问题抽象成动态规划问题

这题是求:用数组 nums 中的数,组成和为 target 的组合个数,顺序不同算不同组合。

  • 总目标 target 可以由 target - num 组合出来;

2. 状态定义

dp[i] 表示组成整数 i 的组合总数(顺序不同算不同)

3. 状态转移方程

目标是 dp[i],它可以由前面所有 dp[i - num] 过来:

  • 如果你想用 num 组成 i,那么 i - num 就必须已经有一个合法组合;
  • 每个 num 都可以尝试添加到组合最后;

所以:dp[i] = dp[i - num1] + dp[i - num2] + ...(只加能让 i - num ≥ 0 的)

4. 初始条件

python
dp[0] = 1 # 因为组成 0 只有一种方法就是什么都不选

5. 目标值

最终答案是 dp[target]

代码实现

python
class Solution:
    def combinationSum4(self, nums, target):
        # 初始化 dp 数组,dp[i] 表示凑出和为 i 的组合数
        dp = [0] * (target + 1)
        # 凑出和为 0 的组合数为 1(即什么都不选)
        dp[0] = 1

        # 遍历从 1 到 target 的每个和
        for i in range(1, target + 1):
            # 尝试使用 nums 中的每个数字
            for num in nums:
                # 如果当前目标和 i 大于等于 num,说明可以用 num 来凑
                if i >= num:
                    # dp[i] 加上使用 num 后剩余和 (i - num) 的组合数
                    dp[i] += dp[i - num]

        # 返回凑出目标和 target 的组合数
        return dp[target]
javascript
/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number}
 */
var combinationSum4 = function(nums, target) {
    const dp = Array(target + 1).fill(0);
    dp[0] = 1;

    for (let i = 1; i < target + 1; i++) {
        for (let num of nums) {
            if (i >= num) {
                dp[i] += dp[i - num];
            }
        }
    }

    return dp[target];
};

复杂度分析

时间复杂度:O(n * target)

空间复杂度:O(target)

链接

377 国际版

377 中文版