二叉树的Morris遍历以及先中后序的实现

Morris算法是一种优化二叉树遍历的方法,将额外空间复杂度降低到O(1)。它通过改变左子树最右节点的右指针来标记节点的访问状态。文章详细介绍了Morris遍历的实现,包括先序、中序和后序遍历的转换,并强调了在使用Morris算法时需要注意的问题,如必须完整执行以保持树结构的正确性。
摘要由CSDN通过智能技术生成

优点

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算法的。
本次总结来源自于左程云老师的算法课。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值