zoukankan      html  css  js  c++  java
  • 刷题总结1:深度优先搜索DFS

    深度优先算法实现伪代码:

    def dfs(i,j):

      #传入搜索坐标

      if 非法条件:

      #判断是否为非法条件,非法条件通常包括横纵坐标超出范围、已搜索过的坐标、不满足题目

        return 0 or False

      grid[i][j] = 0

      #标记访问过的坐标

      for m,n in [[0,1],[0,-1],[1,0],[-1,0]]:

        next_i = i+m

        next_j = j+n

        # 向下、上、右、左接着进行搜索

        ans = dfs(next_i,next_j)

        # 返回答案

      # 视情况而定,需不要将标记还原

    for i in range(m):

      for j in range(n):

        if dfs(i,j):

          return ans

    #进行网格搜索。最终返回需要得到的答案。

    《剑指OFFER》第12题:矩阵中的路径

    请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。

    [["a","b","c","e"],
    ["s","f","c","s"],
    ["a","d","e","e"]]

    但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。

    示例1:

    输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
    输出:true

    思路:本题将利用深度优先搜索和非法情况剪枝。深度优先搜索,即对于满足条件的,一路走到底进行搜索,不满足条件,对上一步进行回溯再进行搜索,是一种升级版的暴力搜索,可以利用非法情况剪枝进行优化,这里的非法情况包括搜索横纵坐标越界、该位置字符与进行对比的给定字符串不相同,判断为False退出。由于不能重复搜索,所以需要对已搜索的格子进行标记。

    代码:

    class Solution:
        def exist(self, board: List[List[str]], word: str) -> bool:
            def dfs(i,j,k):
                #i,j为矩阵中的位置,k为已搜索路径的长度
                if not 0<=i<len(board) or not 0<=j<len(board[0]) or board[i][j] != word[k]:
                    return False
                if k == len(word)-1:
                    return True
                tmp = board[i][j]
                board[i][j] = 0
                #将board[i][j]置为0,防止重复搜索
                if dfs(i,j+1,k+1) or dfs(i,j-1,k+1) or dfs(i+1,j,k+1) or dfs(i-1,j,k+1):
                    return True
                board[i][j] = tmp
                # 如果不满足返回真的条件,则将其还原,开始下一次搜索
            for i in range(len(board)):
                for j in range(len(board[0])):
                    if dfs(i,j,0):
                        return True
            return False

    leetcode 第695题:岛屿的最大面积

    给定一个包含了一些 01 的非空二维数组 grid

    一个 岛屿 是由一些相邻的 1 (代表土地) 构成的组合,这里的「相邻」要求两个 1 必须在水平或者竖直方向上相邻。你可以假设 grid 的四个边缘都被 0(代表水)包围着。

    找到给定的二维数组中最大的岛屿面积。(如果没有岛屿,则返回面积为 0 。)

    示例 1:

    [[0,0,1,0,0,0,0,1,0,0,0,0,0],
     [0,0,0,0,0,0,0,1,1,1,0,0,0],
     [0,1,1,0,1,0,0,0,0,0,0,0,0],
     [0,1,0,0,1,1,0,0,1,0,1,0,0],
     [0,1,0,0,1,1,0,0,1,1,1,0,0],
     [0,0,0,0,0,0,0,0,0,0,1,0,0],
     [0,0,0,0,0,0,0,1,1,1,0,0,0],
     [0,0,0,0,0,0,0,1,1,0,0,0,0]]
    

    对于上面这个给定矩阵应返回 6。注意答案不应该是 11 ,因为岛屿只能包含水平或垂直的四个方向的 1

    思路:这道题同样适用深度优先算法进行搜索,以及非法条件剪枝。如果岛屿为1,则继续进行递归,并且面积更新加一,如果不为1,则面积不加1,最终返回该次搜索的最大面积。非法条件包括搜索横纵坐标越界。该方法同样需要将搜索过的岛屿置为0,以防重复搜索,由于本题求解的是最大面积,并不需要如上一题进行字符串比对,所以并不需要将岛屿还原。

    代码:

    class Solution:
        def maxAreaOfIsland(self, grid: List[List[int]]) -> int:
            def dfs(i,j):
                if not 0<=i<len(grid) or not 0<=j<len(grid[0]) or not grid[i][j]:
                    return 0
                grid[i][j] = 0
                #将走过的路标记为0,防止重复计算
                max_area = 1
                # 最大面积
                for m,n in [[0,1],[0,-1],[1,0],[-1,0]]:
                    next_i = i+m
                    next_j = j+n
                    max_area += dfs(next_i,next_j)
                    # 累计面积
                return max_area
            
            MaxArea = 0
            for i in range(len(grid)):
                for j in range(len(grid[0])):
                    MaxArea=max(dfs(i,j),MaxArea)
            return MaxArea

     《剑指OFFER》面试题13:机器人的运动范围

    地上有一个m行n列的方格,从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?

    示例 1:

    输入:m = 2, n = 3, k = 1
    输出:3

    这道题我们依然使用深度优先算法,不同的是,由于没有给出具体的二维表格,所以需要新建一个m行n列方格来存放是否访问过的标记。所以空间复杂度为O(m*n)。
    增加了一个求数位之和的函数,将其写到判断非法条件语句中,其实现思路与上文的题目相同。机器人从[0,0]坐标开始移动,所以只需要搜索[0,0]的到达路径的格子数即可。
    代码:
    class Solution:
        def movingCount(self, m: int, n: int, k: int) -> int:
            visited = [[False]*n for _ in range(m)]
            #初始化访问坐标的标记
            def sums(x):
                #求数位之和
                s = 0
                while x:
                    s += x % 10
                    x = x // 10
                return s
    
            def dfs(i,j):
                if not 0<=i<m or not 0<=j<n or sums(i)+sums(j) > k or visited[i][j] == True:
                    return 0
                visited[i][j] = True
                #标记访问过的坐标
                ans = 1
                for x,y in [[0,1],[0,-1],[1,0],[-1,0]]:
                    next_i = i+x
                    next_j = j+y
                    ans += dfs(next_i,next_j)
                return ans
            return dfs(0,0)
  • 相关阅读:
    网页的摘要信息
    自我介绍
    DIV和SPAN的区别
    软件工程简介
    设置层的漂移
    构建之法现代软件工程
    手机验证码的相关知识
    Python2和Python3的区别
    我的第一个练习
    结对编程后传之做汉堡
  • 原文地址:https://www.cnblogs.com/amberwang2018/p/12517000.html
Copyright © 2011-2022 走看看