二叉树的遍历我们已经介绍过了最常见的递归遍历:用Java描述数据结构之二叉树,前序遍历,中序遍历,后序遍历
和迭代遍历二叉树前序,中序,后序遍历的迭代实现,实现思路及代码
上面两种遍历虽然看上去不同,但是底层的思路是大同小异的,都是调用栈,递归是用java虚拟机栈,而迭代使用我们自己创建的栈。今天要介绍的莫里斯算法,它的背后没有用任何栈相关的知识,而且相比于前两种空间复杂度为O(n)(因为进栈最差会所有数据全部进栈)的遍历来说,莫里斯算法的空间复杂度只有O(1),这对于量很大的数据来说无疑是天上地下的区别。
Morris遍历
先来简单介绍一下莫里斯算法:
这种方法由 J. H. Morris 在 1979 年的论文「Traversing Binary Trees Simply and Cheaply」中首次提出,因此被称为 Morris 遍历。
介绍就这么一句话,百度百科也没有关于莫里斯算法的介绍,这句话还是复制于力扣。
Morris遍历核心思想
下面介绍莫里斯算法的核心思想:
通过递归和迭代遍历二叉树的核心是回溯,通过栈去回溯已经经过的节点,比如左子树遍历完了,回溯到根节点去遍历右子树,而莫里斯算法对于回溯的实现采用了一种特殊的办法,拿下面这颗树举例:
当把这棵树的根节点传入莫里斯遍历方法中时,cur 指向根节点,算法能回溯的关键是他会创建一个指向next = cur.left(为了将核心思路暂且认为存在left),然后沿着cur.left一路往最右边走next = next.right
当走到最右面时让next.right == cur
这是实现了莫里斯算法回溯的核心,很多人有疑惑为什么是最右边,此时我们把视野放大一点去看
A的左子树它的最右面是H,对于A的左子树的遍历的最后一个节点是H,就是说 cur 遍历完H之后就该回到根节点A了,此时让 cur = cur.right不就完成了回溯,成功回到A了嘛。对于看这篇博客的同学相信已经能理解递归和迭代遍历了,那么也应该能理解我刚刚说的这个是可以应用于递归特性的,即A的左子树的左子树,A的右子树,A的右子树的左子树都适用上面的思想。
介绍完了核心思想,我们来看莫里斯算法的具体代码(注意下面代码中的注释对理解算法也很重要请仔细看):
Morris遍历核心代码
public static void morris(TreeNode root){
//进入方法让cur = root
TreeNode cur = root;
//没有遍历完之前持续循环
while (cur != null){
/*
对于迭代还是递归,一个节点的左子树即使为空
它们都会让cur = left 然后回溯回来
morris这里就会判断如果一个节点的左边为空就不会去
*/
if (cur.left == null){
cur = cur.right;
}else {
/*
让next = cur.left,然后一直往左子树的最右面走
(上文已经说明了为什么要这么做)
*/
TreeNode next = cur.left;
while (next.right != null && next.right != cur)
next = next.right;
/*
此时next经过上面while循环的
next = next.right之后
已经走到了左子树的最右边,我们讨论过此时要让
next.right = cur
以便cur走到这里时可以回溯回去,
这件事是if(next.right == null)时
要做的,else中做的是cur回溯回去了,
但是下一次经历最外面的while循环
时他还会创建一个next = cur.left的引用
然后一直遍历到cur左子树的最右面
这次是第二次经过这里我们知道
next.right != null 而是next.right == cur
为了保证原树不变我们要把这条线擦掉,
即next.right = null,然后让cur = cur.right
即cur的左子树走完了,让cur去走右子树。
*/
if (next.right == null){
next.right = cur;
cur = cur.left;
}else {
System.out.print((char)cur.val + " ");
cur = cur.right;
next.right = null;
}
}
}
}
Morris遍历过程
以上就是莫里斯算法的核心遍历代码,为了方便大家能更加深刻的理解,我们来就例子中的树走一遍该算法(这么良心还不三连):
在感觉还没有理解的情况下请多走两边上图过程,对着代码走最佳。
在介绍了莫里斯算法的核心代码之后