欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

树(一)——Mirrors遍历

程序员文章站 2022-06-15 23:41:03
...

前言

二叉树的遍历是解决二叉树相关问题难以回避的问题,很多问题都是在遍历的基础上来解决的。根据问题的不同,我们可以采用不同的遍历方式。
按照对根节点操作的顺序可以分成:

  1. 先序遍历
  2. 中序遍历
  3. 后序遍历
  4. 层次遍历

按照实现遍历的方式可以分成:

  1. 递归:代码简单,易于理解,但是浪费JVM栈内存。
  2. 迭代:利用辅助结构,实现遍历,需要设计遍历的顺序。

迭代的方式遍历数组的时间复杂度为O(n),空间复杂度为O(h)。h为树的高度。Mirrors遍历的优势在于,不采用辅助结构遍历二叉树,其空间复杂度为O(1)

为什么可以不采用辅助结构??

首先思考一下,为什么我们要用栈或者队列来辅助我们遍历?
因为二叉树的结构是,一个父节点可以轻松的找到子节点,但是子节点无法直接找到父节点。我们需要利用栈或者队列保存访问的记录,以便于我们可以回溯到父节点。
Mirrors遍历采用多个指针,使得通过复杂度不高的操作可以找到父节点。

Mirrors遍历

Mirros遍历是一种节省空间复杂度的方法。将叶子节点上的空指针利用起来,指向父节点,当再次遍历到这个节点的时候再修改回来,这样最后二叉树的结构也没有发生改变。

遍历规则

cur:当前遍历的指针
rightMost:cur节点的左子树的最右节点

  1. 如果cur左子树为空:cur=cur.right;
  2. 如果cur左子树不为空: 找到rightMost
    ——1)如果rightMost.right=null, 那么令rightMost=cur, cur=cur.left;
    ——2)否则,不为空则说明rightMost.right曾经被修改过,我们这是第二次来到这个点,修改rightMost.right=null,cur=cur.right;

解释一下上面的步骤:

  1. 首先整体向左走,如果没有左子树向右走。
  2. 如果左子树不为空,那么就找到左子树上最右的节点这个节点的右子树一定是空的。把这个空指针指向当前节点,即保存了一个指向父节点的指针。此时,左子树还有值没有访问,cur向左走
  3. 3 中的结果是rightMost遍历到这个节点的,cur指针也会遍历到这个节点,这就是第二次访问到了,那么此时该节点作为rightMost时,是被修改过的,所以我们要重新修改其为null。
  4. 当出现3的情况时,说明该节点的左子树全部访问完毕,所以此时cur向右走
  5. cur是真正有效的移动指针,它会走过所有的节点,rightMost是为了修改指针而存在的。

图示

整体上来说是这个样子:如果不考虑指针恢复的操作:树(一)——Mirrors遍历其实不难发现,在这种遍历方式中,有些节点会访问两次,有些节点会访问一次。因为我们有两个指针cur和rightMost,cur会走过所有的节点,rightMost会修改节点指向,使得cur重新访问到某个父节点。
该二叉树的Mirrors序列)(cur走过的顺序)为:
1 2 4 2 5 1 3 6 3 7
树(一)——Mirrors遍历

Code

Mirrors遍历伪代码

 		TreeNode cur = root;
        while (cur != null) {
        // cur指针来遍历二叉树
            TreeNode rightMost = cur;
            // 左子树不为空
            if (cur.left != null) {
            // 向左走,目的是遍历完左子树
                rightMost = cur.left;
                // 找到rightMost
                while (rightMost.right != null && rightMost.right != cur) 
                rightMost = rightMost.right;
                // 查看right指针是否被修改过来
                if (rightMost.right == null) {
                    rightMost.right = cur;
                    cur = cur.left;
                    // 如果没有被修改过,那么说明这个左子树没有遍历完,continue会使得最下面的cur向右移动执行不到,因为我们不希望想右走,左边子树还没有遍历完呢
                    continue;
                } else 
                // 修改过了,这是第二次到了该节点,我们需要将节点的指针修改回来。
                	rightMost.right = null;
            }
            // 如果访问过了,向右走
            cur = cur.right;
        }

基于Mirrors遍历先序

因为所有节点只有俩种情况:访问一次和访问两次
访问一次的节点即没有左子树的节点,那么遇到就打印。
访问两次的节点:有左子树,遇到第一次就打印,第二次不打印

 TreeNode cur = root;
        while (cur != null) {
            TreeNode rightMost = cur;
            if (cur.left != null) {
                rightMost = cur.left;
                while (rightMost.right != null && rightMost.right != cur) rightMost = rightMost.right;
                if (rightMost.right == null) {
                //访问两次的节点:有左子树,遇到第一次就打印,第二次不打印 
                    System.out.println(cur.val);
                    rightMost.right = cur;
                    cur = cur.left;
                    continue;
                } else {
                    rightMost.right = null;
                }
            }
            else
                // 访问一次的节点即没有左子树的节点,那么遇到就打印。
            	System.out.println(cur.val);
            cur = cur.right;

        }

基于Mirrors遍历中序

访问一次的节点即没有左子树的节点,那么遇到就打印。
访问两次的节点:有左子树,遇到第一次不打印,第二次打印

public void inorderTraversal(TreeNode root) {
        TreeNode cur = root;
        while (cur != null) {
            TreeNode rightMost = cur;
            if (cur.left != null) {
                rightMost = cur.left;
                while (rightMost.right != null && rightMost.right != cur) rightMost = rightMost.right;
                if (rightMost.right == null) {
                    rightMost.right = cur;
                    cur = cur.left;
                    // 访问两次的节点:有左子树,遇到第一次不打印,第二次打印,contiue 使得无法执行到打印位置
                    continue;
                } else {
                    rightMost.right = null;
                }
            }
			// 访问一次的节点即没有左子树的节点,那么遇到就打印。
            System.out.println(cur.val);
            cur = cur.right;
        }
        return res;
    }

基于Mirrors遍历后序

后序相对比较复杂。
整体思路还是基于Mirrors遍历。
但是:对于被访问一次的节点,不做任何处理。
访问两次的节点,第二次访问时,逆序打印其左子树右边界。
树(一)——Mirrors遍历我们来看1的左子树,后序访问顺序为 4,4,5,2,按照mirrors访问的顺序来看,当来到4的时候,4的右指针是指向2的,此时,是第二次访问到2节点,我们就逆序打印2的左子树的右边界,此时就到了4,因为4节点是叶子,所以此时输出的内容是4。
接着遍历到下面的4节点,其右指针指向的是1节点。这也是第二次访问到1,逆序打印从2-5-4节点,即1的左子树的右边界。注意此时1并没有打印出来。
到了根节点的右子树也是如此,但是最后无法打印根节点的右边界,所以最后要打印一下,此时也输出了根节点,完全符合后序遍历的结果。
树(一)——Mirrors遍历

     public List<Integer> postorderTraversal(TreeNode root) {
        TreeNode cur = root;
        ArrayList<Integer> res = new ArrayList<>();
        while (cur != null) {
             TreeNode rightMost = cur.left;
             if (rightMost != null) {
                 while (rightMost.right != null && rightMost.right != cur)
                     rightMost = rightMost.right;
                if (rightMost.right == null) {
                    rightMost.right = cur;
                    cur = cur.left;
                    continue;
                } else {
                    rightMost.right = null;
                    function(cur.left,res);
                }
             }
             cur = cur.right;
        }
        function(root,res);
        return res;
    }
    public void function (TreeNode node, ArrayList<Integer> res) {
//        先逆序整个右边界
        TreeNode tail = reverseEdge(node);
        TreeNode cur = tail;
        while (cur != null) {
            res.add(cur.val);
            cur = cur.right;
        }
//        翻转回来
        reverseEdge(tail);
    }
    public TreeNode reverseEdge(TreeNode node) {
        TreeNode pre = null;
        TreeNode next = null;
        while (node != null) {
            next = node.right;
            node.right = pre;
            pre = node;
            node = next;
        }
        return pre;
    }

参考

程序员代码面试指南(第二版)
https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/