力扣整理版八:回溯算法(待更新)

时间:2024-11-21 07:48:36

回溯的本质是穷举,穷举所有可能,然后选出我们想要的答案。

组合/切割/子集/排列/棋盘

1、回溯函数返回值和参数:一般为void

2、终止条件:一般到叶子节点终止

3、回溯的遍历过程:集合的大小树的宽度,递归的深度构成树的深度。

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
        处理节点;
        backtracking(路径,选择列表); // 递归
        回溯,撤销处理结果
    }
}

------------------------------------

(1) 77 组合

(2) 216 组合总和3

(3) 17 电话号码的字母组合

(4) 39 组合总和

(5) 40 组合总和2

------------------------------------

一、组合

1、77 组合

77. 组合 - 力扣(LeetCode)

题目描述:给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。示例: 输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]

------------  python  -------------

未剪枝优化

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        result = []  # 存放结果集
        self.backtracking(n, k, 1, [], result)
        return result
    def backtracking(self, n, k, startIndex, path, result):
        if len(path) == k:
            result.append(path[:])
            return
        for i in range(startIndex, n + 1):  # 需要优化的地方
            path.append(i)  # 处理节点
            self.backtracking(n, k, i + 1, path, result)
            path.pop()  # 回溯,撤销处理的节点

剪枝优化

class Solution:
    def backtracing(self,n,k,index,path,result):
        if len(path)==k:
            result.append(path[:]) # 这里要[:]拷贝当前路径(切片操作),不然后面的操作会更改path path最后输出的都会是[]
            # result.append(path)是对path的引用,path.pop()会改变结果
            return
        for i in range(index,n-(k-len(path))+2):
            path.append(i)
            self.backtracing(n,k,i+1,path,result)
            path.pop()

    def combine(self, n: int, k: int) -> List[List[int]]:
        result=[]
        self.backtracing(n,k,1,[],result)
        return result

2、216 组合总和3

216. 组合总和 III - 力扣(LeetCode)

题目描述:找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

------------  python  -------------

class Solution:
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        result = []  # 存放结果集
        self.backtracking(n, k, 0, 1, [], result)
        return result

    def backtracking(self, targetSum, k, currentSum, startIndex, path, result):
        if currentSum > targetSum:  # 剪枝操作
            return  # 如果path的长度等于k但currentSum不等于targetSum,则直接返回
        if len(path) == k:
            if currentSum == targetSum:
                result.append(path[:])
            return
        for i in range(startIndex, 9 - (k - len(path)) + 2):  # 剪枝
            currentSum += i  # 处理
            path.append(i)  # 处理
            self.backtracking(targetSum, k, currentSum, i + 1, path, result)  # 注意i+1调整startIndex
            currentSum -= i  # 回溯
            path.pop()  # 回溯

3、17 电话号码的字母组合 

17. 电话号码的字母组合 - 力扣(LeetCode)

题目描述:给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。

Tips:对于组合问题如果是一个集合求组合,就需要startindex,如果是多个集合取组合,就不用startindex。

------------  python  -------------

class Solution:
    def letterCombinations(self, digits: str) -> List[str]:
        if not digits:
            return list()
        #如果 digits(s) 是一个空列表 [],那么 ''.join(digits) 会返回一个 空字符串 ""。

        s=[]
        result=[]
        phoneMap = {
            "2": "abc",
            "3": "def",
            "4": "ghi",
            "5": "jkl",
            "6": "mno",
            "7": "pqrs",
            "8": "tuv",
            "9": "wxyz",
        }

        def backtracing(index):
            if index==len(digits):
                result.append("".join(s))
                return
            digit=digits[index]
            for i in phoneMap[digit]:
                # s+=i python中字符串不可以修改,但是列表可以pop()
                s.append(i)
                backtracing(index+1)
                s.pop()

        backtracing(0)
        return result

4、39 组合总和

39. 组合总和 - 力扣(LeetCode)

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

------------  python  -------------

未剪枝

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

        def backtracing(candidates,target,total,startindex,path,result):
            if total>target:
                return
            if total==target:
                result.append(path[:])
                return
            for i in range(startindex,len(candidates)):
                total+=candidates[i]
                path.append(candidates[i])
                backtracing(candidates,target,total,i,path,result)
                total-=candidates[i]
                path.pop()

        backtracing(candidates,target,0,0,[],result)
        return result

        return result

剪枝后

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

        def backtracing(candidates,target,total,startindex,path,result):
            if total==target:
                result.append(path[:])
                return
            # 如果 sum + candidates[i] > target 就终止遍历
            for i in range(startindex,len(candidates)):
                if total+candidates[i]>target:
                    break #剪枝
                total+=candidates[i]
                path.append(candidates[i])
                backtracing(candidates,target,total,i,path,result)
                total-=candidates[i]
                path.pop()

        candidates.sort() #需要排序
        backtracing(candidates,target,0,0,[],result)
        return result

5、40 组合总和2

40. 组合总和 II - 力扣(LeetCode)

题目描述:给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。candidates 中的每个数字在每个组合中只能使用一次。

------------  python  -------------

class Solution:

    def backtracking(self, candidates, target, total, startIndex, path, result):
        if total == target:
            result.append(path[:])
            return

        for i in range(startIndex, len(candidates)):
            if i > startIndex and candidates[i] == candidates[i - 1]:
                continue

            if total + candidates[i] > target:
                break

            total += candidates[i]
            path.append(candidates[i])
            self.backtracking(candidates, target, total, i + 1, path, result)
            total -= candidates[i]
            path.pop()

    def combinationSum2(self, candidates, target):
        result = []
        candidates.sort()
        self.backtracking(candidates, target, 0, 0, [], result)
        return result