NO39 组合总和
题目描述
给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
- 所有数字(包括 target)都是正整数。
- 解集不能包含重复的组合。
1
2
3
4
5
6
7示例 1:
输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
1 | 示例 2: |
这道题很容易联想到递归,我的代码:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23from copy import deepcopy
class Solution:
def combinationSum(self, candidates, target):
if not candidates:
return []
result=[]
candidates.sort()
def combinate(src,obj,target):
src_tmp=deepcopy(src)
if min(src)>target:
return
for i in src:
if src_tmp.index(i)==1:
src_tmp.pop(0)
if i==target:
obj.append(i)
result.append(obj)
elif i<target:
obj.append(i)
combinate(src_tmp,deepcopy(obj),target-i)
obj.pop()
combinate(candidates, [],target)
return result
排名第一的算法是这样的:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16class Solution:
def combinationSum(self, candidates, target):
candidates.sort()
res = []
def combination(candidates, result, target):
for idx, i in enumerate(candidates):
if i == target:
res.append(result + [i])
return
if i < target:
combination(candidates[idx:], result + [i], target - i)
else:
break
return
combination(candidates, [], target)
return res