Leetcode 39. 组合总和

Leetcode 39. 组合总和

题目说明

给定一个无重复元素的数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明：

所有数字（包括 target）都是正整数。
解集不能包含重复的组合。
示例 1：

输入：candidates = [2,3,6,7], target = 7,
所求解集为：
[
[7],
[2,2,3]
]
示例 2：

输入：candidates = [2,3,5], target = 8,
所求解集为：
[
[2,2,2,2],
[2,3,3],
[3,5]
]

题目解析

1. 回溯 + 剪枝

   创建 caculate 函数，传参为( 已加入元素列表，剩余总和，还剩元素列表 )。
   为了防止返回列表重复，传参需要传入还剩元素列表。

2. 时间复杂度 O(n log n ) ，空间复杂度 O( n )

Python代码

class Solution:
    def combinationSum(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 sum - remain_l[i] < 0:
                    return
                elif sum - remain_l[i] == 0:
                    result.append(l + [remain_l[i]])
                else:
                    caculate(l + [remain_l[i]], sum - remain_l[i], remain_l[i:])

        caculate([], target, candidates)
        return result