回溯法与枚举法

最新推荐文章于 2022-03-03 14:24:10 发布

MuBaicao

最新推荐文章于 2022-03-03 14:24:10 发布

阅读量2.3k

点赞数 3

分类专栏：面试文章标签：回溯法

本文链接：https://blog.csdn.net/woshicao11/article/details/82024120

版权

面试专栏收录该内容

10 篇文章 1 订阅

订阅专栏

1 回溯法应用

回溯法，顾名思义，生活中有一类人，很执拗，比如我，不到黄河心不死，到了黄河怎么办？往回走呗，难不成跳下去？

回溯法解决的问题有全排列，全组合，枚举什么的。

比如，给定一串不重复字母，让你按字典顺序进行排列，这是个全排列的问题，很好理解，但编程很麻烦。如果给定的字母可以重复，让你按字典的顺序排列，这个问题是不是更复杂了呢？

回溯法，跟数一样，随机取出一个元素，此时情况有N种，则有N个这有的节点，然后在剩下的集合中，又随机取个数，此时情况可能有N-1种（可能小于N-1如果有重复的项），重复上面的生成节点的过程。当处理到最后一个元素或者当前的路径以及不满足要求了，就往回走了，递归回来。

回溯构建数的时候，取出了那个元素，下一个集合就没有这个元素了。

2 回溯法固化模板

回溯法就相当于搜索一棵树，使用深度优先搜索数的方法。

def dfs(输入):
    if 当前状态为边界：
        记录或输出
        return 

    for i in range(len(输入))：#横向遍历所有子节点
        目前遍历的当前子节点
        if 子状态满足约束条件：
            dfs(子状态)

3 题集

3.1 字符串排列（全排列问题）

题目描述

输入一个字符串,按字典序打印出该字符串中字符的所有排列。例如输入字符串abc,则打印出由字符a,b,c所能排列出来的所有字符串abc,acb,bac,bca,cab和cba。

解题思路：字典排序，重复元素的全排列。深度优先搜索，即为空是返回．先从Ｎ个样本中取一个，接在递归在剩下的样本中取，为什么要递归？因为这样再取完以后会退回到上一级．

重复原始处理，排好序后，如果取到的这个，和上个相似，就continue．

假设根就是空，第一层会有N个子节点，对重复的取一个X，然后从集合中去掉取出的这一个X（重复的也去掉一个），迭代。

class Solution:
    def Permutation(self, ss):
        if not ss:
            return []
        ss = sorted(ss)
        temp = []
        N = len(ss)
        def dfs(ss,cur):
            if not ss:
                if len(cur) == N:
                    temp.append(cur)
                return
        for i in range(len(ss)):
            if i > 0:
                if ss[i] == ss[i-1]:
                    continue
            if len(ss) == N:
                cur = ss[i]
            else:
                cur = cur+ss[i]
            if i+1 < len(ss) and i>0:
                ss1 = ss[:i]+ss[i+1:]
            elif i==0:
                ss1 = ss[i+1:]
            else:
                ss1 = ss[:i]
            dfs(ss1, cur)
        dfs(ss,'')
        return temp

3.2 矩阵中的路径

题目描述

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始，每一步可以在矩阵中向左，向右，向上，向下移动一个格子。如果一条路径经过了矩阵中的某一个格子，则之后不能再次进入这个格子。例如 a b c e s f c s a d e e 这样的3 X 4 矩阵中包含一条字符串"bcced"的路径，但是矩阵中不包含"abcb"路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

解题思路：首先输入的是字符串‘ csadee’，需要映射成矩阵（嗯，绕了很长时间）。因为起始位置是随机的，所以需要遍历所有的矩阵作为起始位置。为了节约时间，我们需要剪枝，即如果当前位置的值不等于path，那就可以直接返回。

还需要注意一个约束，就是不能往回走，最开始我的代码是：

class Solution:
    def __init__(self):
        self.flag = False
    def hasPath(self, matrix, rows, cols, path):
        if not matrix:            
            return False
        for j in range(rows):
            for i in range(cols):
                self.hasp(matrix,j,i,path[:], rows)
                if self.flag:
                    return True
        return False

    def hasp(self,matrix, curr, curc, path, rows, cols):
        if not path:
            self.flag = True
            return
        if curr<0 or curc<0 or curr>rows-1 or curc>cols-1:
            return
        index = curr * cols + curc
        if matrix[index] == path[0]:
              
            self.hasp(matrix, curr-1, curc, path[1:], rows, cols)
            
            self.hasp(matrix, curr + 1, curc, path[1:], rows, cols)
            
            self.hasp(matrix, curr , curc - 1, path[1:], rows, cols)
            
            self.hasp(matrix, curr , curc + 1, path[1:], rows, cols)

本地跑属于False的也会返回True，什么原因呢？因为我重复了曾经的路径，贴上正确的代码：

class Solution:
    def __init__(self):
        self.flag = False
    def hasPath(self, matrix, rows, cols, path):
        if not matrix:            
            return False
        for j in range(rows):
            for i in range(cols):
                self.hasp(matrix,j,i,path[:], rows, cols,[(j,i)])
                if self.flag:
                    return True
        return False

    def hasp(self,matrix, curr, curc, path, rows, cols,history):
        if not path:
            self.flag = True
            return
        if curr<0 or curc<0 or curr>rows-1 or curc>cols-1:
            return
        index = curr * cols + curc
        if matrix[index] == path[0]:
            if (curr-1, curc) not in history:            
                self.hasp(matrix, curr-1, curc, path[1:], rows, cols,history+[(curr-1, curc)])
            if (curr + 1, curc) not in history: 
                self.hasp(matrix, curr + 1, curc, path[1:], rows, cols,history+[(curr + 1, curc)])
            if (curr , curc-1) not in history: 
                self.hasp(matrix, curr , curc - 1, path[1:], rows, cols,history+[(curr, curc-1)])
            if (curr , curc+1) not in history: 
                self.hasp(matrix, curr , curc + 1, path[1:], rows, cols,history+[(curr , curc+1)])

3.2 矩阵中的路径

题目描述

地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于k的格子。例如，当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。但是，它不能进入方格（35,38），因为3+5+3+8 = 19。请问该机器人能够达到多少个格子？

解题思路：数位之和，我想到的就是把它转换成str，然后按位读取str，转换成int并求和。感觉这道题很简单啊，思路也对，时间复杂度总是超了。我原来的思路跟上道题一样，边界条件改了（超出矩阵和数位之和），其实条件就只为（0,0）。调了很久，后来知道是走过的路径应该设置为全局变量。当初没有设置为全局变量。

正确代码：

class Solution:
    def movingCount(self, threshold, rows, cols):
        self.row, self.col = rows, cols
        self.dict = []
        
        self.search(threshold, 0, 0)
        return len(self.dict)

    def search(self, threshold, i, j):
        ss = 0
        for k in str(i)+str(j):
            ss = ss+int(k)
        if ss > threshold:
            return
        self.dict.append((i,j))
        if i != self.row - 1 and (i+1,j) not in self.dict:
            self.search(threshold, i + 1, j)
        if j != self.col - 1 and (i,j+1) not in self.dict:
            self.search(threshold, i, j + 1)