【LeetCode】40组合总数

题目描述

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

题目理解

本题主要是在上一道题（组合总数）的基础上增加了重复的问题，所以在上一个的搜索树的基础上增加剪枝就可以完成该功能了。

剪枝的方式如下：对于当前层，相同的数字我只取一个。原有的剪枝：对于当前数字和已经超过数字target的情况，直接剪枝。

代码（py）

import copy
res = []
class Solution:
    def combinationSum2(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        res.clear()
        if len(candidates) == 0:
            return res
        candidates.sort()
        self.combinationSumHelper([],-1,target,candidates)
        return res
    def combinationSumHelper(self,now,i,tar,candidates):
        if sum(now) == tar and now not in res:
            dummy = copy.deepcopy(now)
            res.append(dummy)
            return
        if sum(now) > tar:
            return
        else:
            temp = -9999
            for j in range(i+1,len(candidates)):
                if candidates[j]!=temp:
                    now.append(candidates[j])
                    self.combinationSumHelper(now,j,tar,candidates)
                    now.pop()
                    temp = candidates[j]