题目描述
给定一个无重复元素的数组 candidates
和一个目标数 target
,找出 candidates
中所有可以使数字和为 target
的组合。
candidates
中的数字可以无限制重复被选取。
说明:
- 所有数字(包括
target
)都是正整数。 - 解集不能包含重复的组合。
示例 1:
输入: candidates =[2,3,6,7],
target =7
, 所求解集为: [ [7], [2,2,3] ]
示例 2:
输入: candidates = [2,3,5],
target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]
题目理解
本题目主要是为了取数据,在 已有的数组里面可以取多次,最后取得的数据之和等于目标值。这个问题在我前不久面试小米的时候刚好遇见过,当时我的最初想法是用动态规划思路来解决,后来面试官跟我讲动态规划对于求解这种需要过程的结果不是很好,我就想到了利用回溯的思路解决这个问题。
解决思路,首先对所有的数据排序,然后由每一个数据向下发散,每个数据只能发散比自己更大的数,这样就避免了重复。如果遇到发散之后的数据和大于目标值,说明已经过了,可以考虑剪枝。
代码(py)
import copy
res = []
class Solution:
def combinationSum(self, candidates, target):
"""
:type candidates: List[int]
:type target: int
:rtype: List[List[int]]
"""
res.clear()
if len(candidates) == 0:
return res
candidates.sort()
self.combinationSumHelper([],0,target,candidates)
return res
def combinationSumHelper(self,now,i,tar,candidates):
if sum(now) == tar:
dummy = copy.deepcopy(now)
res.append(dummy)
return
if sum(now) > tar:
return
else:
for j in range(i,len(candidates)):
now.append(candidates[j])
self.combinationSumHelper(now,j,tar,candidates)
now.pop()
记得用一个全局的res来存储结果,利用Helper来辅助递归。