【算法】图论中DFS和BFS模板讲解

图论的解题模板和二叉树基本一致,都是在DFS和BFS基础上进行求解。

二叉树的DFS和BFS模板如下所示:

复制代码
public void DFSTree(TreeNode root){
  if(root==null)
      return null;
  DFSTree(root.left);
  DFSTree(root.right);
}

public void BFSTree(TreeNode root){
  if(root==null) return;
  Deque<TreeNode> queue=new ArrayDeque<>();
  queue.offer(root);
  while(!queue.isEmpty()){
      TreeNode curNode=queue.poll();
      if(curNode.left!=null){
          queue.offer(curNode.left);
      }
      if(curNode.right!=null){
          queue.offer(curNode.right);
      }
  }
}

根据二叉树的DFS和BFS模板可以发现,整代码有两个要素:访问相邻结点和判断base case。

由于二叉树本身是一个递归的定义,一颗二叉树它的左右子树也是一颗二叉树,因此访问相邻结点只需访问左右结点即可。

而对DFS来说,base case就是root==null,这有两层含义:一是递归的终止条件,表示当前root指向的结点为空,要停止递归,不需要再往下遍历了。二是root==null表示遍历到了叶子结点,应及时返回,避免出现root.leftroot.right空指针异常。而对BFS来说,base case就是while(!queue.isEmpty())curNode.left!=nullcurNode.right!=nullwhile(!queue.isEmpty())表示当前队列不为空,即还需要继续遍历,而curNode.left!=nullcurNode.right!=null则保证入队的元素均不为空。

类比二叉树的特性,可以写出图中DFS和BFS的两个要素:1、图中某个格子的相邻结点是什么?该如何访问?2、判断base case。

对于格子(row,col)来说,它的相邻结点分别在上下左右,分别为(row-1,col)、(row+1,col)、(row,col-1)、(row,col+1),其次类比二叉树通过root==null来充当base case确保不需要继续遍历,那么在图中的base case应该就是出现数组越界的格子,这样就能得到图论中DFS和BFS的基础架构

复制代码
public void DFSGrid(int[][] grid,int row,int col){
    // 判断 base case
    if(row<0||col<0||row>=grid.length||col>=grid[0].length)
      return;
    // 访问上、下、左、右四个相邻结点
    DFSGrid(grid,row-1,col);
    DFSGrid(grid,row+1,col);
    DFSGrid(grid,row,col-1);
    DFSGrid(grid,row,col+1);
}

public void BFSGrid(int[][] grid,int row,int col){
  Deque<int[]> queue=new ArrayDeque<>();
  queue.offer(new int[]{row,col});
  //处理当前格子等操作
  grid[row][col]=已遍历
  while(!queue.isEmpty()){
      int[] cur=queue.poll();
      row=cur[0];
      col=cur[1];
      //判断base case
      if(row>=0&&col>=0&&row<grid.length&&col<grid[0].length){
      //记录当前格子等操作
      grid[row][col]=已遍历
      // 访问上、下、左、右四个相邻结点
      queue.offer(new int[]{row-1,col});
      queue.offer(new int[]{row+1,col});
      queue.offer(new int[]{row,col-1});
      queue.offer(new int[]{row,col+1});
      }
  }
}

在不熟悉的情况下,可能会疑惑为什么DFS(grid,row,col)和queue.offer(grid,new int[]{row-1,col})时不用判断参数是否合法?难度不会数组越界吗?其实可以先合法性判断,然后再进行DFS和BFS,但是我们用的方法是先使用后判断,将判断的操作统一放在前面,这样在取上下左右四个位置的时候不用判断四次参数的合理性,而是先将其视为合法的进行保存,然后在要访问的时候在统一判断合法性,避免了多次冗余的判断。

其次对于BFS的queue中为什么要存坐标,是因为通过坐标可以直接访问到对应的格子,同时更改坐标可以快速定位到其上下左右四个格子,若存储格子,那如何传递参数来访问上下左右四个格子呢?

相关推荐
运器12325 分钟前
【一起来学AI大模型】支持向量机(SVM):核心算法深度解析
大数据·人工智能·算法·机器学习·支持向量机·ai·ai编程
Zedthm34 分钟前
LeetCode1004. 最大连续1的个数 III
java·算法·leetcode
lcg_magic42 分钟前
图论系列(一):基础概念与术语解析
图论
神的孩子都在歌唱1 小时前
3423. 循环数组中相邻元素的最大差值 — day97
java·数据结构·算法
YuTaoShao1 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法一)空间复杂度 O(M + N)
算法·leetcode·矩阵
dying_man2 小时前
LeetCode--42.接雨水
算法·leetcode
vortex52 小时前
算法设计与分析 知识总结
算法
艾莉丝努力练剑3 小时前
【C语言】学习过程教训与经验杂谈:思想准备、知识回顾(三)
c语言·开发语言·数据结构·学习·算法
ZZZS05163 小时前
stack栈练习
c++·笔记·学习·算法·动态规划
hans汉斯3 小时前
【人工智能与机器人研究】基于力传感器坐标系预标定的重力补偿算法
人工智能·算法·机器人·信号处理·深度神经网络