【java】剑指offer27_二叉树的镜像

最新推荐文章于 2021-09-26 17:08:33 发布

ic_xcc

最新推荐文章于 2021-09-26 17:08:33 发布

阅读量86

点赞数

分类专栏：算法 java 文章标签：二叉树算法 java

本文链接：https://blog.csdn.net/ic_xcc/article/details/113248369

版权

算法同时被 2 个专栏收录

128 篇文章 0 订阅

订阅专栏

java

122 篇文章 0 订阅

订阅专栏

题目描述

请完成一个函数，输入一个二叉树，该函数输出它的镜像。

例如输入：

     4
   /   \
  2     7
/ \   / \
1   3 6   9
镜像输出：

     4
   /   \
  7     2
/ \   / \
9   6 3   1

示例 1：

输入：root = [4,2,7,1,3,6,9]
输出：[4,7,2,9,6,3,1]

限制：

0 <= 节点个数 <= 1000

解题思路一：递归，判断root是否为空，暂存左节点，遍历二叉树将起左节点重新赋值为相应右节点，左节点赋值为暂存的右节点值

递归解析：

终止条件：当节点 root 为空时（即越过叶节点），则返回 null ；
递推工作：

初始化节点 tmp ，用于暂存 root 的左子节点；
开启递归右子节点mirrorTree(root.right) ，并将返回值作为root 的左子节点。
开启递归左子节点 mirrorTree(tmp) ，并将返回值作为 root 的右子节点。

返回值：返回当前节点 root ；

Q：为何需要暂存 rootroot 的左子节点？
A：在递归右子节点 “root.left = mirrorTree(root.right);root.left=mirrorTree(root.right);” 执行完毕后， root.leftroot.left 的值已经发生改变，此时递归左子节点 mirrorTree(root.left)mirrorTree(root.left) 则会出问题。

public TreeNode mirrorTree(TreeNode root) {
        if(root == null){
            return null;
        }
        TreeNode temp = root.left; //暂存二叉树做节点
        root.left = mirrorTree(root.right);//左节点更改为右节点的值
        root.right = mirrorTree(temp);//右节点更改为之前暂存左节点的值
        return root;
    }

复杂度分析：
时间复杂度 O(N) ：其中 N 为二叉树的节点数量，建立二叉树镜像需要遍历树的所有节点，占用 O(N) 时间。
空间复杂度 O(N) ：最差情况下（当二叉树退化为链表），递归时系统需使用 O(N) 大小的栈空间

参考思路二：迭代，将二叉树的节点存放在栈中，利用栈记录所有的节点，再交换所有的左右节点

算法流程：

特例处理：当 root 为空时，直接返回 null ；
初始化：栈（或队列），本文用栈，并加入根节点 root 。
循环交换：当栈 stack 为空时跳出；

出栈：记为 node ；
添加子节点：将 node 左和右子节点入栈；
交换：交换 node 的左 / 右子节点。

返回值：返回根节点 root 。

public TreeNode mirrorTree(TreeNode root) {
        if(root == null){
            return null;
        }
        Stack<TreeNode> stack = new Stack<>(); //新建栈
        stack.add(root);//将root节点放入栈
        while (!stack.isEmpty()){
            TreeNode node = stack.pop();//栈弹出一个节点
            //将节点不为空的左右节点放入栈
            if(node.left != null){
                stack.add(node.left);
            }
            if(node.right != null){
                stack.add(node.right);
            }
            //暂存左节点--交换左右节点
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;
        }
        return root;
    }