0040. 组合总和 II

0040. 组合总和 II #

  • 标签:数组、回溯
  • 难度:中等

题目大意 #

给定一个数组 candidates 和一个目标数 target,找出 candidates 中所有可以使数字和为目标数 target 的组合。

数组 candidates 中的数字在每个组合中只能使用一次,且 1 ≤ candidates[i] ≤ 50

解题思路 #

跟「 0039. 组合总和」不一样的地方在于本题不能有重复组合,所以关键步骤在于去重。

在回溯遍历的时候,下一层递归的 start_index 要从当前节点的后一位开始遍历,即 i + 1 位开始。而且统一递归层不能使用相同的元素,即需要增加一句判断 if i > start_index and candidates[i] == candidates[i - 1]: continue

代码 #

 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
26
27
class Solution:
    res = []
    path = []
    def backtrack(self, candidates: List[int], target: int, sum: int, start_index: int):
        if sum > target:
            return
        if sum == target:
            self.res.append(self.path[:])
            return

        for i in range(start_index, len(candidates)):
            if sum + candidates[i] > target:
                break
            if i > start_index and candidates[i] == candidates[i - 1]:
                continue
            sum += candidates[i]
            self.path.append(candidates[i])
            self.backtrack(candidates, target, sum, i + 1)
            sum -= candidates[i]
            self.path.pop()

    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        self.res.clear()
        self.path.clear()
        candidates.sort()
        self.backtrack(candidates, target, 0, 0)
        return self.res
本站总访问量  次  |  您是本站第  位访问者