[Leetcode]组合总和II-Python递归

[Leetcode]组合总和II-Python递归

Leetcode-组合总和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]
]

解题思路

• 这道题与Leetcode-组合总和的区别在于，candidates 中的每个数字在每个组合中只能使用一次，那么就要考虑去重，排序之后可以比较一下是否与上一个数字相同，相同则跳过

实现代码

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates.sort()
        n = len(candidates)
        res = []

        def helper(i,cur_sum,ans):
            if cur_sum == target:
                res.append(ans)
                return
            if i == n or cur_sum > target:
                return            
            for j in range(i, n):
                if j!=i and candidates[j] == candidates[j-1]: //如果重复则跳过
                    continue
                if cur_sum + candidates[j] > target:
                    break
                helper(j + 1, cur_sum + candidates[j], ans + [candidates[j]])

        helper(0, 0, [])
        return res