LeetCode-Python-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]
]

 

思路：

带剪枝的回溯DFS解决。

为了避免输出重复需要加一个排序去重的步骤。

class Solution(object):
    def combinationSum2(self, c, t):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        res = list()
        l = len(c)
        def dfs(start, tmp):
            s = sum(tmp)
            if s == t:#满足题目要求
                tt = sorted(tmp)
                if tt not in res:#避免重复
                    res.append(tt[:])
                return
            if start >= l:
                return

            for i in range(start, l):
                
                if c[i] > t or s + c[i] > t:#剪枝
                    continue
                
                tmp.append(c[i])
                dfs(i + 1, tmp) #下一层
                tmp.pop()#回溯
                    
        for i,x in enumerate(c):
            dfs(i, list())
            
        return res