39典型回溯 2020-09-09(未允禁转)

2020-09-09  本文已影响0人  9_SooHyun

39. 组合总和

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

说明:
所有数字(包括 target)都是正整数
解集不能包含重复的组合

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

思路1

回溯

这是一道典型的回溯问题,dfs回溯思路如下

# 回溯
class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        res = []
        candidates.sort()
        visited_list = []
        def backTracking(candidates, target, visited_list):
            if target == 0:
                res.append(visited_list[:])
            if target < 0:
                return 
            for i in range(len(candidates)):
                visited_list.append(candidates[i])
                new_target = target - candidates[i]
                # dfs
                backTracking(candidates[i:], new_target, visited_list)
                # 状态恢复
                visited_list.pop()
        
        backTracking(candidates, target, visited_list)
        return res

思路2

递归做法

### 递归做法 ###
# 对于排序后的candidates,从左到右遍历一次,取出第i个元素,更新target值为new_target
# 这时问题转换为了规模更小的同类问题,即candidates[i:]和new_target下的问题
# 那么就可以考虑用递归的方式不断缩小target的值,
# 我们要保证递归函数helper接受的参数target必须大于等于0。如果target为0,到达递归出口

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates.sort()
        
        # 用递归做
        def helper(candidates, target):
            # 递归出口
            res = []
            if target == 0:
                res.append([])
                return res
            
            for i in range(len(candidates)):
                ele_took = candidates[i]
                new_target = target - ele_took
                # 剪枝
                if new_target < 0:
                    break
                for l in helper(candidates[i:], new_target):
                    res.append([ele_took] + l)
            return res
        
        return helper(candidates, target)

小结

【带返回值的递归】是,【从递推的角度看待问题】,只关心下一规模问题的解,假设得到了下一规模问题的解,推导当前规模问题的解就迎刃而解,本质思想和dp是一致的--不同规模问题的状态转移,如果再保存中间结果就是dp的逆过程了

回溯是【不带返回值的递归】,它从【宏观的整体的角度】看待问题,把一个问题所有可能的解走完,并在来到叶子结点时利用全局变量完成结算。回溯是直接解决整一个问题的思路,而不是解决一个个同类子问题来简化当前问题的思路

上一篇 下一篇

猜你喜欢

热点阅读