算法系列-二叉树遍历(非递归实现)

简介: 在内卷潮流的席卷下,身为算法小白的我不得不问自己,是否得踏上征程,征服这座巍巍高山。从零开始,终点不知何方,取决于自己可以坚持多久。希望你可以和我一样,克服恐惧,哪怕毫无基础,哪怕天生愚钝,依然选择直面困难。

在内卷潮流的席卷下,身为算法小白的我不得不问自己,是否得踏上征程,征服这座巍巍高山。

从零开始,终点不知何方,取决于自己可以坚持多久。

希望你可以和我一样,克服恐惧,哪怕毫无基础,哪怕天生愚钝,依然选择直面困难。

数据结构分类

  • 队列

前言

本篇作为二叉树遍历的下篇,我们主要分析如何通过非递归的方式实现二叉树的遍历。

有对二叉树不了解或单纯想知道二叉树的遍历实现的同学,建议先阅读算法系列-二叉树遍历

二叉树的遍历

概念不多说,我们直接进入算法学习。

二叉树的第一个算法就是遍历算法,而遍历又分为深度优先算法,广度优先算法。其中深度优先算法又分为前序,中序,后序优先。

  • 深度优先遍历
  • 前序遍历
  • 中序遍历
  • 后序遍历
  • 广度优先遍历

下文的代码我们将以下面的二叉树为例,先用树结构来描述它。

93.png



const tree = {
  value: 'A',
  left: {
    value: 'B',
    left: {
      value: 'D',
    },
    right: {
      value: 'E'
    }
  },
  right: {
    value: 'C',
    left: {
      value: 'F',
    },
    right: {
      value: 'G'
    }
  }
}
复制代码

深度优先遍历


深度优先即先访问子节点,直到叶子结点。

我们称深度优先遍历为 DFS(Deep First Search) 深度优先搜索。


前序遍历


前序遍历的前序是相对根节点来说的,即访问顺序为

根节点 -> 左节点 -> 右节点


此时,我们的访问顺序依次为

A -> B -> D -> E -> C -> F -> G

const preOrder = (node) => {
  const stack = [node]
  while(stack.length) {
    const node = stack.pop()
    console.log(node.value)
    if (node.right) {
      stack.push(node.right)
    }
    if (node.left) {
      stack.push(node.left)
    }
  }
}
复制代码


解释一下


我们借用了数据结构,先访问根节点,再往栈内推入右节点,左节点,在下一轮循环中则是利用出栈先访问左节点。


中序遍历


中序遍历的中序是相对根节点来说的,即访问顺序为


左节点 -> 根节点 -> 右节点


此时,我们的访问顺序依次为

D -> B -> E -> A -> F -> C -> G

const inOrder = (node) => {
  const stack = [node];
  while(stack.length) {
    const node = stack.pop()
    if (node.visited) {
      console.log(node.value)
      delete node.visited
      continue
    }
    node.visited = true
    if (node.right) {
      stack.push(node.right)
    }
    stack.push(node)
    if (node.left) {
      stack.push(node.left)
    }
  }
}
复制代码


中序遍历我我们同样使用了结构,与前序遍历相比其难点在于我们无法在遇到当前节点时就访问其值,因为我们必须先访问左节点。所以我们这边利用了额外的属性 visited 来标记其,再次访问时则直接访问其值则可。


刚开始的时候一直想不出如何绕过根节点的访问,因为自己觉得不应该往节点添加数据。后面看了别人的实现才发现也是通过额外属性标记实现了,所以也效仿了别人的做法。


当然,后面我又找到了没有引入额外属性的实现方式,果然还是自己太愚蠢了,没有想到 node.leftnode.right 的转化。

const inOrder = (node) => {
  const stack = [node];
  while(stack.length) {
    // 先遍历左节点
    if (node) {
      stack.push(node)
      node = node.left
    } else {
      node = stack.pop()
      console.log(node.value)
      // 重点在于这 将node转移到右节点
      node = node.right
    }
  }
}
复制代码


此算法的利用了左节点同时为根节点的特性,使得我们完成了 左节点 -> 根节点 的遍历,后面在根节点处又使用了 node = node.right 将当前遍历转移到右节点,非常巧妙,这大概就是我想破头没想出来的原因吧!


后序遍历


后序遍历的后序是相对根节点来说的,即访问顺序为

左节点 -> 右节点 -> 根节点


此时,我们的访问顺序依次为

D -> E -> B -> F -> G -> C -> A

const postOrder = (node) => {
  const stack = [node]
  while(stack.length) {
    const node = stack.pop()
    if (node.visited) {
      delete node.visited
      console.log(node.value)
      continue
    }
    node.visited = true
    stack.push(node)
    if (node.right) {
      stack.push(node.right)
    }
    if (node.left) {
      stack.push(node.left)
    }
  }
}
复制代码


实现逻辑和中序遍历相似,这边就不重复解释了


同样,我们来看看另外一种不加额外属性的实现方式

const postOrder = (node) => {
  const s1 = [node]
  const s2 = []
  while (s1.length) {
    node = s1.pop()
    s2.push(node)
    if (node.left) {
      s1.push(node.left)
    }
    if (node.right) {
      s1.push(node.right)
    }
  }
  while(s2.length) {
    console.log(s2.pop().value)
  }
}
复制代码


此解法利用了两个来实现后序遍历是我万万想不到的,但是如果仔细分析可以发现实际和先序遍历的实现有些相似。


是的,它们本来就是比较相似的两种顺序,先序是 根节点 -> 左右子节点,后序是 左右子节点 -> 根节点


所以多加分析可以发现 s1 实际是模仿先序的实现,s2 则用于反转访问。也就是 s1

实现栈递归,s2 实现访问节点存储。


PS:u1s1,让我想真想不出来


广度优先遍历


相对而言,广度优先遍历会麻烦一些,我们称其为 BFS(Breath First Search) 广度优先搜索。


广度优先遍历也叫层次遍历,即访问顺序为一层一层地访问。


在本例子中,访问先后顺序为 A -> B -> C -> D -> E -> F -> G

const BFS = (nodes) => {
  for (let i = 0; i < list.length; i++) {
    const curNode = list[i]
    const { left, right, value } = curNode
    console.log(value);
    if (left) {
      list.push(left)
    }
    if (right) {
      list.push(right)
    }
  }
}
复制代码


利用迭代实现广度优先反而比深度优先来的简单一些,原理就是利用数据结构队列,先遍历当前层,再将下层节点加入队列中,最后实现按层按顺序遍历。


总结


本篇文章作为上篇文章的续篇,通过代码演示了如何通过迭代的方式遍历二叉树,与递归实现相比,会显得稍微复杂一些。好吧,实际不止一些。


既然递归简单,并且听起来也比较高大上一些,那我们为什么还要研究迭代方式实现呢,其原因在于递归的性能实际是比较差的,感兴趣的朋友们可以去网上搜索相关知识了解下。


对于二叉树的算法实际还有很多,比如根据遍历结构来生成二叉树,也许我们下篇文章会对二叉树的生成算法进行研究。


参考





相关文章
|
1月前
|
存储 算法
【数据结构与算法】二叉树基础OJ--下(巩固提高)
【数据结构与算法】二叉树基础OJ--下(巩固提高)
|
1月前
|
机器学习/深度学习 算法
递归算法题练习(数的计算、带备忘录的递归、计算函数值)
递归算法题练习(数的计算、带备忘录的递归、计算函数值)
|
1月前
|
算法 Java
算法:Java计算二叉树从根节点到叶子结点的最大路径和
算法:Java计算二叉树从根节点到叶子结点的最大路径和
|
24天前
|
算法
【算法与数据结构】二叉树(前中后)序遍历2
【算法与数据结构】二叉树(前中后)序遍历
|
1月前
|
机器学习/深度学习 数据采集 监控
机器学习-特征选择:如何使用递归特征消除算法自动筛选出最优特征?
机器学习-特征选择:如何使用递归特征消除算法自动筛选出最优特征?
72 0
|
1月前
|
存储 缓存 算法
【数据结构与算法】【小白也能学的数据结构与算法】递归 分治 迭代 动态规划 无从下手?一文通!!!
【数据结构与算法】【小白也能学的数据结构与算法】递归 分治 迭代 动态规划 无从下手?一文通!!!
|
15天前
|
算法
算法系列--递归(一)--与链表有关(上)
算法系列--递归(一)--与链表有关
27 0
|
1天前
|
算法
数据结构与算法-递归思想
数据结构与算法-递归思想
6 0
|
10天前
|
算法 DataX
二叉树(中)+Leetcode每日一题——“数据结构与算法”“剑指Offer55-I. 二叉树的深度”“100.相同的树”“965.单值二叉树”
二叉树(中)+Leetcode每日一题——“数据结构与算法”“剑指Offer55-I. 二叉树的深度”“100.相同的树”“965.单值二叉树”
|
19天前
|
存储 算法 搜索推荐
【数据结构与算法】归并排序(详解:递归与非递归的归并排序 | 赠:冒泡排序和选择排序)
【数据结构与算法】归并排序(详解:递归与非递归的归并排序 | 赠:冒泡排序和选择排序)