Leetcode 40. 组合总和 II

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. 回溯 + 剪枝

   （1） 对列表排序

   （2）caculate 函数 传参（ 已加入列表元素，剩余和，未加入列表元素）

   （3）剪枝：若元素与之前元素相同，continue；若剩余元素的第一个元素大于剩余和，结束递归

Python代码

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        result = []
        candidates = sorted(candidates)

        def caculate(l, sum, remain_l):
            for i in range(len(remain_l)):
                if i > 0 and remain_l[i] == remain_l[i - 1]:
                    continue
                if sum - remain_l[i] < 0:
                    return
                elif sum - remain_l[i] == 0:
                    result.append(l + [remain_l[i]])
                else:
                    if i + 1 < len(remain_l):
                        caculate(l + [remain_l[i]], sum - remain_l[i], remain_l[i + 1:])

        caculate([], target, candidates)
        return result