好久没做题了,更菜了5555
题目描述
给你一个由 不同 整数组成的数组 nums ,和一个目标整数 target 。请你从 nums 中找出并返回总和为 target 的元素组合的个数。
题目数据保证答案符合 32 位整数范围。
实例
输入: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)
请注意,顺序不同的序列被视作不同的组合。
思路
注意:题目要求的是不同顺序也算一种组合
比较容易想到的方法是DFS,但是亲测超时了,这里就不加赘述。
动态规划
建立一个长度为target(目标)+1的数组,记录每个i(0<=i<=target)的值有多少组合。
如果目标是0,那么dp[0]=1,一个数字都不选。
可以先将nums数组排序,做两层for循环,外层是1 ~ target,内层是0 ~ len(nums)-1。
dp[i]=dp[i-nums[0]]+dp[i-nums[1]]+dp[i-nums[2]]+……+dp[i-nums[j]] (nums[j+1]>i)
代码
class Solution:
def combinationSum4(self, nums: List[int], target: int) -> int:
nums.sort()
dp=[0]*(target+1)
dp[0]=1
for i in range(1,len(dp)):
for j in range(len(nums)):
if nums[j]>i:
break
else:
dp[i]+=dp[i-nums[j]]
return dp[-1]