[LeetCode]40、组合总和2

2019-07-28  本文已影响0人  河海中最菜

题目描述

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

思路解析

只能用一次,就是不要继续重复使用当前元素,在上一题的基础加了一点的改动。

class Solution:
    def combinationSum2(self, candidates, target):
        res = []
        if not candidates or len(candidates) == 0:
            return res
        candidates.sort()

        def helper(temp, temp_sum, i):
            if temp_sum == target:
                if temp not in res:
                    res.append(temp)
                return
            if temp_sum > target or i >= len(candidates):
                return
            helper(temp + [candidates[i]], temp_sum + candidates[i], i + 1)
            helper(temp, temp_sum, i + 1)
        helper([], 0, 0)
        return res

    def combinationSum2_(self, candidates, target):
        if not candidates or len(candidates) == 0:
            return []
        candidates.sort()
        res = []
        def helper(temp, temp_sum, index):
            if temp_sum == target:
                res.append(temp)
                return
            for i in range(index, len(candidates)):
                # 判断当前元素是否能够加入组合
                if temp_sum + candidates[i] > target:
                    break
                # 避免重复元素,并不是只一个元素能够重复多次的那种
                if i > index and candidates[i] == candidates[i-1]:
                    continue
                helper(temp + [candidates[i]], temp_sum + candidates[i], i+1)
        helper([], 0, 0)
        return res
AC40
上一篇下一篇

猜你喜欢

热点阅读