回溯法-组合总和 II

题目描述

给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。

说明：

• 所有数字（包括目标数）都是正整数。
• 解集不能包含重复的组合。

示例

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为: [ [1, 7], [1, 2, 5], [2, 6], [1, 1, 6] ]

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为: [ [1,2,2], [5] ]

分析

利用回溯法子集树模板，注意这里的判断要大量用到求和，所以利用模板中的01映射会花费大量的时间，这里不用01映射，再加个全局的和变量记录，降低每次求和带来的时间消耗。

事实上模板用的01映射只是帮助理解问题，理解后皆可取消模板内的01映射。

代码

class Solution:
    def __init__(self):
        self.data = []
        self.n = 0
        self.tempRes = []
        self.res = []
        self.target = 0
        self.sumNum = 0 
    

    def backtrack(self, k):
        if k>=self.n or self.sumNum >= self.target: 
            if self.sumNum == self.target:
                self.res.append(self.tempRes[:])
                
        else:
            for i in [0, 1]:
                if i==1:
                    data = self.data[k]
                    self.tempRes.append(data)
                    self.sumNum += data
 
                self.backtrack(k+1)
                if i==1:
                    self.tempRes.pop()
                    self.sumNum -= data

    def combinationSum2(self, candidates, target):  
        self.data = sorted(candidates)
        self.n = len(candidates)
        self.target = target

        self.backtrack(0)
        res = list(map(list,set(map(tuple, self.res))))
        return res