39. 组合总和![](https://i-blog.csdnimg.cn/blog_migrate/0b48b3b32206cf5c7182dd79a73d4a27.png)
这道题的思路和我们之前做的一样,不过唯一不同的地方就是可以不在意答案元素内是否有重复的元素,之前我们的startindex是从i+1开始每次递归,那么这里既然元素可以取多次,那么我们的下次递归依旧是从i开始,就不是i+1了。
递归三部曲:
循环终止条件:
当target_sum==target的时候,就加入到答案数组中
递归参数:startindex,candidates,target,target_sum
单层递归逻辑:当前元素加入并且判断之后,就要进行剪枝操作,减掉当前元素并且pop出去。
代码:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
res = []
path = []
def dfs(target_sum,nums,target,start):
if target_sum>target:
return
if target_sum==target:
res.append(path[:])
return res
for i in range(start,len(nums)):
target_sum+=nums[i]
path.append(nums[i])
dfs(target_sum,nums,target,i)
target_sum-=nums[i]
path.pop()
dfs(0,candidates,target,0)
return res
40. 组合总和 II
思路:本题核心难度在于 数组内的元素可以重复,但是数组是不能重复的,比如[1,7]和[7,1]只能取一个。
我们转化为树形结构来看,有两个维度,一个是树的同一层,另一个是树的分支,也就是说树的分是可以取重复的,但是在树的同一层中,之前用过的就不能再用第二次了。
如图所示:
强调一下,树层去重的话,需要对数组排序!
回溯三部曲:
递归参数:这里还是设置一个index,然后一个sum来记录求和
递归终止条件:如果sum+candidates[I]>target,注意我们的数组已经是排好序了的,那么如果开头加上就已经大于target了,后面的也只会更大,就不用再继续了。
单层循环逻辑:如果循环到当前元素,判断是否和前面的元素一样,并且这里的i是大于0的哈。如果和前面元素一样的话continue,就相当于是跳过了本层的重复元素。
代码:
def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
res = []
path = []
def backtracking(candidates,total_sum,target,start_index):
candidates.sort()
if total_sum==target:
res.append(path[:])
return res
for i in range(start_index,len(candidates)):
if total_sum + candidates[i] > target:
return
if i >start_index and candidates[i]==candidates[i-1]:
continue
total_sum+=candidates[i]
path.append(candidates[i])
backtracking(candidates,total_sum,target,i+1)
total_sum-=candidates[i]
path.pop()
backtracking(candidates,0,target,0)
return res