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

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class Solution:
    def combinationSum2(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        result = []
        if not candidates:
            return result
        candidates.sort()
        self.dfs(candidates, target, 0, [], result)
        return result
    
    def dfs(self, candidates, target, index, current, result):
        while index < len(candidates):
            if candidates[index] > target:
                return
            elif candidates[index] == target:
                result.append(current + [candidates[index]])
                return 
            self.dfs(candidates, target - candidates[index], index + 1, current + [candidates[index]], result)
            index += 1
            while index < len(candidates) and candidates[index] == candidates[index-1]:
                index += 1