题目
给你一个由 不同 整数组成的数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iv
思路
状态表示
用一个一维的状态dp[i]
表示目标和为
i
i
i的时候解的个数,则dp[target]
即为所求。
状态转移
初始状态dp(0)=1
,因为和为0只有”所有数都不取“这一种方案。
对于任意的i.j
,i
的解加上nums[j]
即为i+nums[j]
的解的子集
将i+nums[j]
记为x
,则x
的解即为所有满足i+nums[j] = x
的i
的解之和
也就是dp[x] = dp[x-nums[0]] + dp[x-nums[1]] + ... + dp[x-nums[n]]
知道这一点,我们就可以从dp[0]
一直递推到dp[target]
可能说的有点绕,用示例1来举个例子:
代码实现
对于每个状态dp[i]
,遍历一次nums
,累加dp[i-nums[j]]
到dp[i]
当遍历nums
时,就保证了nums
中的每一个元素都有机会作为解排列的最后一位(而且是从i=1开始就是如此),进而保证了所有顺序不同的组合都能被考虑到。
class Solution {
public int combinationSum4(int[] nums, int target) {
int[] dp = new int[target + 1];
dp[0] = 1;
for (int i = 1; i <= target; i++) {
for (int num : nums) {
if (i >= num) {
dp[i] += dp[i - num];
}
}
}
return dp[target];
}
}