数据结构与算法|遍历二叉树的神级方法-Morris遍历【建议收藏】数据结构|深度学习|二叉树|算

在前面，我们简单提及过二叉树的遍历方式，有递归和非递归两个版本的遍历。仔细想一想，不管是递归的，还是非递归的遍历，两种版本的遍历都是需要耗费大量的、额外的空间。比如当我们二叉树的高度有100层，那么递归时，系统就会一直压栈，最坏情况下，一直要压入100次遍历的递归函数，因为此处的空间复杂度是跟这颗二叉树的高度相关的。所以有人就在想，有没有什么方式，能够使这个空间复杂度再压缩一点呢？

文章图片

前期文章：二叉树的非递归遍历
本期文章源码：GitHub
有，肯定是有的。那就是今天的主题：Morris遍历。这种思想就能使二叉树的遍历，空间复杂度为O(1)。那我们直接开始吧！

文章目录

一、Morris
二、前序遍历
三、中序遍历
四、后序遍历（较难）

一、Morris 在讲解之前，我们来分析一下。为什么遍历二叉树需要压栈？

文章图片

观察上面这颗二叉树，脑补一下。当我们一直往下遍历时，走到值为4的节点，打印输出4后，我该怎么回到2节点？每个节点都是只有指向左右孩子的引用，并没有指向父节点的引用。
所以在遍历4节点的时候，会提前将2节点的所有信息，放入栈里面，只有在4节点遍历完成之后，才会弹出栈里的信息（2节点）。然后继续遍历其他的节点。
这也就是为什么遍历二叉树需要压栈的原因。
那么我们就在想啊，有没有种方法，能够不压栈，就能从4节点直接返回到2节点呢？那就是利用4节点的右孩子引用。因为4节点是叶子节点，没有左右孩子节点的。我们将4节点的右孩子引用，指向2节点，这样就能够从4节点直接返回到2节点了。具体的图片如下：

文章图片

上图橙色的线条，是Morris方法的作用，就是改变某些节点的右孩子引用。然后在具体的遍历过程中，我们将橙色线条擦除即可。保证跟原二叉树一模一样。这就是Morris的思想。那么问题来了，我们该如何画这些橙色的线条呢？
前提：我们准备两个引用变量：cur和mostRight。 cur表示当前遍历的节点；mostRight表示cur节点左子树中，往右边遍历，第一个没有右孩子的节点。举个例子：上图中cur是1节点，则mostRight就是1节点左子树中，往右下角遍历，5节点就是第一个没有右孩子的节点。
Morris遍历细节：（cur初始化为根结点）

如果cur没有左孩子，cur就向右孩子移动。（cur = cur.right）
如果cur有左孩子，那就找到cur左子树中，最靠右的节点mostRight：
- 如果mostRight的右孩子是null，说明是第一次遍历到mostRight。此时让mostRight的右孩子指向cur。
- 如果mostRight的右孩子是cur，说明是第二次遍历到mostRight。此时让mostRight的右孩子等于null即可。
遍历结束条件：cur等于null时

伪代码：

public void morris(Node root) {if (root == null) {return; }Node cur = root; Node mostRight = null; while (cur != null) {mostRight = cur.left; if (mostRight 不等于null) {1、循环找到mostRight节点 2、判断mustRight的右孩子是否为null }//mostRight等于null时 cur = cur.right; //向右子树移动 } }

上面的伪代码就是大致的框架结构，具体的代码如下：

文章图片

值得注意的是，当第一次遍历到mostRight时，让mostRight的右孩子指向cur之后，cur 再往左子树走，然后应该再写一句continue。让代码继续往cur的左子树继续遍历。
可能有的同学分不清，什么时候是第一次遍历到mostRight，什么是第二次遍历到mostRight。简单点说，第一次就是mostRight的右孩子是null时，此时我们需要将右孩子指向cur；第二次是mostRight的右孩子指向cur的时候，此时我们需要将右孩子重新改回来，改为null。

关于前序、中序、后序遍历，都是在Morris的基础之上，添加printf即可。具体的，请往下看。
二、前序遍历我们都知道前序遍历的顺序是：头左右。
而Morris改前序遍历，非常简单。记住两个点，你就能独自改前序遍历。

如果cur节点没有左孩子，那就打印当前cur的值
如果是第一次遍历到mostRight节点，那就打印当前cur的值

就以上两个点，添加到代码里面即可。

文章图片

其他的代码，原封不动。只添加这两句代码即可完成前序遍历。
三、中序遍历前序遍历顺序：左头右。
中序跟前序的遍历，这里很相似，也是记住两个点即可：

如果cur没有左孩子，直接打印输出cur的值
如果mostRight是第二次遍历到，那就打印输出cur的值

跟前序遍历的区别只是mostRight这里，前序是第一次遍历到mostRight就打印cur，中序是第二次遍历到就打印。

文章图片

当然，这两个printf可以提取出来，写一行即可。这里只是为了让大家清晰地认识到中序遍历。
四、后序遍历（较难） Morris改后序遍历，稍微有一点点难，因为Morris遍历出来的左右结果，都满足一个规律：当前节点没有左子树，那么只会遍历一次这个节点；当前节点有左子树，那么会有mostRight节点会返回到cur节点，也就是说有左孩子的节点，会遍历两次。
后序遍历是每个节点，遍历到第3次的时候，才打印输出，现在可好，Morris遍历最多只能遍历到2次，那该如何是好？？？
不急，我们来看一张图：

文章图片

不难发现，红色的数值就是后序遍历的结果。对比下左边的二叉树，红色线条的指向，从左到右，从下到上的打印顺序，竟然跟后序遍历的结果是一样的。
那我们就以这个为切入点，当遍历cur的时候，我们就可以打印cur左子树的最靠右的那一排的节点。比如cur等于1节点的时候，我们就可以打印2、5节点。
要满足从下到上的打印顺序，最容易想到的就是搞一个栈，压栈进行，然后再依次弹出栈并打印。这样的话，空间复杂度就不是O(1)了。大家是否还记得逆序单链表？我们将那一排的节点，进行逆序，然后打印输出之后，再逆序回来即可。

//逆序那一排节点 public Node reverseList(Node node) {Node pre = null; Node next = null; while (node != null) {next = node.right; //逆序右子树那一排节点 node.right = pre; pre = node; node = next; } return pre; }

【数据结构与算法|遍历二叉树的神级方法-Morris遍历【建议收藏】】逆序完成之后，打印输出即可，然后在逆序回来，保持原来的状态

public void printList(Node node) {Node newHead = reverseList(node); //逆序 node = newHead; //先保存newHead，等会逆序while (newHead != null) {System.out.print(newHead.val + " "); newHead = newHead.right; //向右子树走 } reverseList(node); //逆序回来，保持原状态 }

上面两个方法，就能实现打印行为。现在的问题是，怎么进行调用？？？
Morris遍历，有的节点会遍历到一次，而有的节点会遍历到两次。牢牢抓住这两种情况，Morris遍历就简单了。
这里的后序遍历，还是跟前序和中序差不多，只是打印函数，我们在第二次遍历到mostRight调用即可。
比如：在上图中，第二次遍历到mostRight等于4节点时，我们就调用打印函数，传递的参数是cur的左孩子。再比如：第二次遍历到mostRight等于5节点时，此时调用printList（cur.left），此时的cur是1节点。
看代码更清晰：