[Leetcode]组合总和-Python递归

[Leetcode]组合总和

Leetcode-组合总和

题目描述

给定一个无重复元素的数组 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 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每个元素都是独一无二的。
1 <= target <= 500

解题思路

• 递归回溯
• 递归终止条件：
  • 超出数组范围
  • 当前所选的结果和大于target
  • 当前所选的结果和等于target
• 递归调用：
  • 选择当前数进行下一轮递归
  • 跳过当前数进行下一轮递归

实现代码

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        res=[]
        def helper(i,cur_sum,ans):
            if i == len(candidates) or cur_sum > target:
                return
            if cur_sum ==target:
                res.append(ans)
                return
            helper(i, cur_sum + candidates[i], ans + [candidates[i]]) #选择当前数
            helper(i + 1, cur_sum , ans) #跳过当前数
        helper(0, 0, [])
        return res
            

• 方法二

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        res=[]

        def helper(i,cur_sum,ans):
            if i == len(candidates) or cur_sum > target:
                return
            if cur_sum ==target:
                res.append(ans)
                return
            for j in range(i, len(candidates)): //类似于迭代的调用递归                
                helper(j, cur_sum + candidates[j], ans + [candidates[j]])   
                
        helper(0, 0, [])
        return res

• 利用排序

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates.sort()  //排序
        res=[]

        def helper(i,cur_sum,ans):
            if i == len(candidates) or cur_sum > target:
                return
            if cur_sum ==target:
                res.append(ans)
                return
            for j in range(i, len(candidates)):
                if cur_sum + candidates[j] >target: //排序之后，如果已经大于target就不用继续递归后面
                    break
                helper(j, cur_sum + candidates[j], ans + [candidates[j]])   
                
        helper(0, 0, [])
        return res