加起来和为目标值的组合2-回溯

加起来和为目标值的组合2-牛客

思路

• 回溯法可解决的问题
  • 组合问题：N个数里面按一定规则找出k个数的集合
  • 切割问题：一个字符串按一定规则有几种切割方式
  • 子集问题：一个N个数的集合里有多少符合条件的子集
  • 排列问题：N个数按一定规则全排列，有几种排列方式
  • 棋盘问题：N皇后，解数独等等
    本题属于子集问题
    这类寻找所有可行解的题，尝试用搜索来解，本题用深度优先搜索dfs。
• 回溯
  1、入参和返回值：入参：目标值targe，开始下标i; 返回值：无
  2、终止条件：target==0，表示已经找到满足和为target元素.
  3、本层任务：去重后，把元素加入tmp_res
  注意：递归结束，要把tmp_res中的元素释放掉，再开始下一次循环。

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param num int整型一维数组 
# @param target int整型 
# @return int整型二维数组
#
class Solution:
    def combinationSum2(self , num: List[int], target: int) -> List[List[int]]:
        #回溯用dfs+去重+剪枝
        res=[]
        tmp_res=[]
        if len(num)==0:
            return res
        num=sorted(num)
        if len(num)==1 and num[0] > target:
            return res
        def backtrack(target, start):
            if target <= 0:
                if target==0:
                    res.append(tmp_res[:])
                return
            for i in range(start, len(num)):
                #去重
                if i > start and num[i]==num[i-1]:
                    continue
                tmp_res.append(num[i])
                #递归
                backtrack(target-num[i], i+1)
                tmp_res.pop()
        backtrack(target, 0)
        return res
                

回溯模板

res=[]
path=[]
 
def dfs(start,path,num,res):
#start-初始位置，path-搜索路径，num-原数组，res-最终结果
    if 结束条件
    	#浅拷贝path
        res.append(path[:])
        return
    
    for i in range(len(num)):
        path.append(num[i])
        dfs(i+1,path,num,res)
        #append后，地址没变，需要pop，撤销回溯
        path.pop()