优点
morris算法可以使二叉树遍历的时候,额外的空间复杂度从O(M)变为O(1)。其中M为树的高度
算法的实现
实现代码
public static void morris(Node head) {
if (head == null) {
return;
}
//就只有这些额外空间复杂度
Node cur = head;
Node mostRight = null;
while (cur != null) {
//找是否有左孩子
mostRight = cur.left;
//如果有左孩子则找左孩子的最右节点,mostRight.right != cur,这句话确保不会出现循环找
if (mostRight != null) {
while (mostRight.right != null && mostRight.right != cur) {
mostRight = mostRight.right;
}
//此时mostRight为左孩子的最右节点
if (mostRight.right == null) {
mostRight.right = cur;
//然后cur向左移动
cur = cur.left;
continue;
} else { //如果不为空则说明mostRight的右孩子为cur
// 这时要将其右指针变为null
mostRight.right = null;
//cur向右移动,这一步一起合并到了下数3行
}
}
cur = cur.right;
}
}
遍历顺序
任何有左树的节点都会被遍历2次,并且2次节点之间也夹着这个左节点,否则一次。
本质
本质:左树上的最右节点的右指针状态来标记,到底是第几次到达该节点( 分1次/2次)
遍历方式
先序
将morris序变为先序:节点第一次到达时处理,第二次到达时忽略。
public static void morrisPre(Node head) {
if (head == null) {
return;
}
//就只有这些额外空间复杂度
Node cur = head;
Node mostRight = null;
while (cur != null) {
//找是否有左孩子
mostRight = cur.left;
//如果有左孩子则找左孩子的最右节点,mostRight.right != cur,这句话确保不会出现循环找
if (mostRight != null) {
while (mostRight.right != null && mostRight.right != cur) {
mostRight = mostRight.right;
}
//此时mostRight为左孩子的最右节点
if (mostRight.right == null) {
//第一次来到时打印
System.out.println(cur.value);
mostRight.right = cur;
//然后cur向左移动
cur = cur.left;
continue;
} else { //如果不为空则说明mostRight的右孩子为cur
// 这时要将其右指针变为null
mostRight.right = null;
//cur向右移动,这一步一起合并到了29行
}
} else {
//没有左树直接打印
System.out.println(cur.value);
}
cur = cur.right;
}
}
中序
将morris序变为中序:能2次遍历的节点(有左树的节点),节点第二次到达时处理,第1次到达时忽略。没有左树的节点直接打印。本质和右移步骤相同。
后序
将morris序变为后序:能2次遍历的节点(有左树的节点),第二次到达时,逆序打印左树的右边界。最后逆序打印整个树的右边界。
但如何逆序打印右边界?需要将节点的右指针逆序。
public static void morrisPos(Node head) {
if (head == null) {
return;
}
//就只有这些额外空间复杂度
Node cur = head;
Node mostRight = null;
while (cur != null) {
//找是否有左孩子
mostRight = cur.left;
//如果有左孩子则找左孩子的最右节点,mostRight.right != cur,这句话确保不会出现循环找
if (mostRight != null) {
while (mostRight.right != null && mostRight.right != cur) {
mostRight = mostRight.right;
}
//此时mostRight为左孩子的最右节点
if (mostRight.right == null) {
//第一次来到时打印
System.out.println(cur.value);
mostRight.right = cur;
//然后cur向左移动
cur = cur.left;
continue;
} else { //如果不为空则说明mostRight的右孩子为cur
// 这时要将其右指针变为null
mostRight.right = null;
printEdge(cur.left);
//cur向右移动,这一步一起合并到了29行
}
}
cur = cur.right;
}
printEdge(head);
}
private static void printEdge(Node head) {
//右边界逆序
Node tail = reverseEdge(head);
Node cur = tail;
while (cur != null) {
System.out.println(cur.value);
cur = cur.right;
}
//再逆序回来
reverseEdge(tail);
}
private static Node reverseEdge(Node head) {
Node pre = null;
Node next = null;
while (head != null) {
next = head.right;
head.right = pre;
pre = head;
head = next;
}
return pre;
}
Morris算法求解时的问题
注意,使用Morris算法判断是否是什么样的二叉树时,一定要全都做完再返回,如果中途返回,可能会导致树的结构出现问题。
当一个算法需要左树的信息+右树的信息+自己的信息后才能做决定,这种需要3次回到自己才能解决的算法是不能用Morris算法的。
本次总结来源自于左程云老师的算法课。