leetcode组合总和 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）
链接：https://leetcode-cn.com/problems/combination-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

和上一道组合总和基本一样，只是有重复数字，另外每个数字只能用一次
所以在回溯的时候加上两个条件，一是每次判断candidates[i]==candidates[i-1]的话就跳过，因为在[i,r]区间内的解肯定包括了[i+1,r]区间内的解；另一个不同就是每次回溯时因为不能重复取，所以区间[i,r]会变成[i+1,r]本数只会取一次

代码

class Solution:
    def backtrack(self, ans, cur, candidates, target,l):
        if target == 0:
            ans.append(list(cur))
            return
        if target < candidates[0]:
            return
        r = len(candidates)
        for i in range(l, r):
            num = candidates[i]
            if i != l and candidates[i] == candidates[i-1]:
                continue
            target -= num
            cur.append(num)
            self.backtrack(ans, cur, candidates, target, i+1)
            cur.remove(num)
            target += num
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        ans = []
        cur = []
        candidates = sorted(candidates)
        self.backtrack(ans, cur, candidates, target, 0)
        return ans