题目描述
二叉树的遍历,无论是递归版本还是非递归版本,都需要借助自己写的栈或者系统栈来实现,那么,空间复杂度就不可能是O(1),并且,递归版本中,每个节点都要被遍历三次,时间复杂度也相对较高。有能够实现时间复杂度O(N),空间复杂度O(1)的算法来遍历二叉树吗?
Morris遍历
Morris遍历是利用二叉树叶子节点中的大量空指针,来实现向上回溯的算法,从而达到节省空间的目的。其实就是借助线索二叉树的思想。
回想一下经典二叉树的遍历,递归时的回溯,其实是系统栈帮我们保存了所有中间过程,然后通过压栈出栈实现的,那么,我们是否可以利用叶子节点的空指针,来实现回溯呢?这就是Morris遍历的实质。
Morris遍历流程:
开始时,从头节点位置。
1.如果当前cur节点没有左孩子,则向右移动:cur=cur.right
2.如果当前cur节点有左孩子,则找到这个左孩子的最右节点mostRight。
-----2.1如果mostRight的右指针为空,则让他指向cur,然后cur向左移动:cur=cur.left
-----2.2如果mostRight的右指针指向cur,则重置为null,然后cur向右移动:cur=cur.right
cur为空时,遍历结束
假如二叉树为上述结构,那么Morris遍历的结果为:
1、2、4、2、5、1、3、6、3、7
可以看到,叶子节点只会被遍历一次,而非叶子节点会被遍历两次,但是递归的话,每个节点会被遍历三次,因此,Morris遍历的时间复杂度为O(N),且比递归遍历的常数项小,空间复杂度为O(1)。
代码
public static void morris(Node node){
if(node==null){
return;
}
while (node!=null){
System.out.println(node.value);
if(node.left!=null){
Node mostRight=node.left;
while (mostRight.right!=null&&mostRight.right!=node){
mostRight=mostRight.right;
}
if(mostRight.right==null){
mostRight.right=node;
node=node.left;
continue;
}else {
mostRight.right=null;
}
}
node=node.right;
}
}
但是,有人会说,这遍历有重复值嘛,不是前序中序后序遍历,当然,这只是Morris遍历的最初形态,我们可以在此基础上改成前序遍历、中序遍历、后序遍历,同时时间复杂度和空间复杂度不变。
Morris改前序遍历
Morris遍历的结果上,所有节点第一次出现就打印,第二次出现不打印,那么,前面的那个例子就变成了:1、2、4、5、3、6、7
这不就是前序遍历嘛
Morris改中序遍历
在Morris遍历的基础上,如果是叶子节点,只会被遍历一次,那就直接打印,如果不是叶子节点,会被遍历两次,那就第二次再打印,那么,前面的例子就变成了:4、2、5、1、6、3、7
这不就是中序遍历嘛
Morris改后续遍历
Morris遍历只能最多回到自己两次,无法像递归那样三次,如何实现后续遍历呢?
叶子节点只能回到自己一次,什么都不做,能回到自己两次的节点,在第二次回到本身的时候,就逆序打印他左树的右边界。当整棵树遍历完成后,最后,逆序打印整棵树的右边界,Morris改后续遍历就实现了。
注意:逆序遍历这一步,也要实现空间复杂度O(1),不能使用栈,其实也容易,因为是右边界嘛,一直向右,其实相当于一个单链表,其实也就是逆序打印单链表,打印完再恢复。