leetcode 40. 组合总和 II (python)

Posted 2022-06-21 xiaotongtt

给定一个数组 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]
]

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        if len(candidates)<=0:
            return []
        candidates.sort()
        self.res = []
        self.dfs(candidates,[],target,0)
        return self.res
    def dfs(self,candidates,sublist,target,index):
        if target == 0:
            self.res.append(sublist)
            return
        if target < 0:
            return
        for i in range(index,len(candidates)):  # 从当前index开始遍历
            if i != index and candidates[i] == candidates[i-1]: # 保证每个数字只使用一次
                continue
            self.dfs(candidates,sublist+[candidates[i]],target-candidates[i],i+1)

参考链接： https://blog.csdn.net/weixin_40546602/article/details/88357837

回溯递归函数dfs(self,candidates,sublist,target,index），index为当前遍历到数组的位置，因为不允许重复使用元素，只可依次遍历。

剪枝操作1： 当target值已经小于0，由于数组中包含负数，所有之间返回

剪枝操作2：保证每个数字只使用一次