回溯+剪枝.40.组合总和II

题目描述

给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。
说明: 所有数字(包括目标数)都是正整数。 解集不能包含重复的组合。
示例 1:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为: [ [1, 7], [1, 2, 5], [2, 6], [1, 1, 6] ]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为: [ [1,2,2], [5] ]

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/combination-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

回溯模板套上,加上剪枝条件,因为每个数字只能使用一次,所有需要去重。
想到的方法就是,将 candidate 进行排序,重复的数字必定连续。
得出剪枝条件1:candidates[i] == candidates[i-1]
第二个剪枝条件就是‘和’超过target,
得出剪枝条件1:sums+candidates[i] > target

代码

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        self.res = []
        candidates.sort()
        def reversive(sums, start, tmp):
            for i in range(start, len(candidates)):
                if i > start and candidates[i] == candidates[i-1]:  # 剪枝1, 每个重复数字只计算一次
                    continue
                if sums+candidates[i] > target:  # 剪枝2, 超出临界条件
                    break 
                if sums+candidates[i] == target: # 结果
                    self.res.append(tmp+[candidates[i]])
                    break 
                reversive(sums+candidates[i], i+1, tmp+[candidates[i]])
        reversive(0, 0, [])
        return self.res

复杂度分析

  • 时间复杂度:O(n*2^n)
  • 空间复杂度:O(target)

你可能感兴趣的:(leetcode刷题笔记,python,算法,回溯,剪枝)