给定一个无重复元素的数组 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]
]
思路:
回溯法。
#自己写的丑陋版本,很慢……
class Solution(object):
def combinationSum(self, candidates, target):
"""
:type candidates: List[int]
:type target: int
:rtype: List[List[int]]
"""
res = list()
def generate(c, t, tmp, s):
# print s
if s == target:
# print tmp
res.append(tmp[:])
return
if s > t:
return
for digit in c:
s = sum(tmp) + digit
tmp.append(digit)
generate(c, t, tmp, s)
tmp.pop()
generate(candidates, target, [], 0)
#----以下为去重
for i in range(0, len(res)):
res[i].sort()
ress = list()
for i in range(0, len(res)):
flag = 0 # 1 重复 0 单独
for j in range(i + 1, len(res)):
if res[i] == res[j]:
flag = 1
break
if not flag:
ress.append(res[i])
return ress
#大神写的
class Solution(object):
def combinationSum(self, candidates, target):
res = []
candidates.sort()
def backtrack(remain, temp, start):
if not remain: #remain为0
res.append(temp[:])
else:
for i, n in enumerate(candidates[start:]):
if n > remain:
break
backtrack(remain-n, temp+[n], start+i)
backtrack(target, [], 0)
return res