-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path39_Combination_Sum.py
More file actions
97 lines (78 loc) · 3.2 KB
/
39_Combination_Sum.py
File metadata and controls
97 lines (78 loc) · 3.2 KB
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
'''
39. Combination Sum
Medium
16.3K
328
Companies
Given an array of distinct integers candidates and a target integer target, return a list of all unique combinations of candidates where the chosen numbers sum to target. You may return the combinations in any order.
The same number may be chosen from candidates an unlimited number of times. Two combinations are unique if the
frequency
of at least one of the chosen numbers is different.
The test cases are generated such that the number of unique combinations that sum up to target is less than 150 combinations for the given input.
Example 1:
Input: candidates = [2,3,6,7], target = 7
Output: [[2,2,3],[7]]
Explanation:
2 and 3 are candidates, and 2 + 2 + 3 = 7. Note that 2 can be used multiple times.
7 is a candidate, and 7 = 7.
These are the only two combinations.
Example 2:
Input: candidates = [2,3,5], target = 8
Output: [[2,2,2,2],[2,3,3],[3,5]]
Example 3:
Input: candidates = [2], target = 1
Output: []
Constraints:
1 <= candidates.length <= 30
2 <= candidates[i] <= 40
All elements of candidates are distinct.
1 <= target <= 40
'''
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
result = []
current = []
currentSum = 0
self.backtrack(candidates,target,result, current,currentSum,0)
return result
def backtrack(self, candidates: List[int], target: int, result: List[int], current: List[int] ,currentSum: int, startindex: int) -> List[List[int]]:
curr = current.copy()
if currentSum==target:
return result.append(curr)
elif currentSum>target:
return
else:
for i in range(startindex, len(candidates)):
current.append(candidates[i])
self.backtrack(candidates,target,result,current,currentSum+candidates[i],i) # not i+1 because we are reusing the candidates[i]
current.remove(candidates[i])
'''Approach: Backtracing
1. Same like subset (78. Subsets) except we are reusing the candidates value and finding all solutions.'''
#Updated:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
res=[]
current=[]
def backtrack(target,i):
if target==0:
res.append(current[:])
return
elif target<0 or i==len(candidates):
return
current.append(candidates[i])
backtrack(target-candidates[i],i) # not i+1 because we are reusing the candidates[i]
current.pop()
backtrack(target,i+1)
backtrack(target,0)
return res
# Updated approach:
# 1. Use Backtracking function
# if target is zero then append the current list to result
# if target is less than zero then return
# else iterate through the candidates list
# append the current candidate to the current list
# call the backtracking function with target-candidates[i] and i
# pop the last element from the current list
# call the backtracking function with target and i+1
# return the result list
# Time complexity: O(2^n)
# Space complexity: O(n)