/
39.py
61 lines (49 loc) · 1.44 KB
/
39.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# 39. Combination Sum
# Given a set of candidate numbers (candidates) (without duplicates) and
# a target number (target), find all unique combinations in candidates
# where the candidate numbers sums to target.
# The same repeated number may be chosen from candidates unlimited number of times.
# Note:
# All numbers (including target) will be positive integers.
# The solution set must not contain duplicate combinations.
# Example 1:
# Input: candidates = [2,3,6,7], target = 7,
# A solution set is:
# [
# [7],
# [2,2,3]
# ]
# Example 2:
# Input: candidates = [2,3,5], target = 8,
# A solution set is:
# [
# [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]]
"""
if target < 0:
return []
if target == 0:
return [[]]
results = []
for candidate in candidates:
for result in self.combinationSum(candidates, target - candidate):
tmp = sorted(result + [candidate])
if tmp not in results:
results.append(tmp)
return results
if __name__ == "__main__":
from util import Test
s = Solution()
t = Test(s.combinationSum)
r1 = [[2, 2, 3], [7]]
t.equal(r1, [2, 3, 6, 7], 7)
r2 = [[2, 2, 2, 2], [2, 3, 3], [3, 5]]
t.equal(r2, [2, 3, 5], 8)