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

763 阅读5分钟

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

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

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

数据结构分类

  • 队列

前言

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

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

二叉树的遍历

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

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

  • 深度优先遍历

    • 前序遍历

    • 中序遍历

    • 后序遍历

  • 广度优先遍历

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

屏幕快照 2021-12-18 上午11.06.15.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)
    }
  }
}

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

总结

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

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

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

参考