力扣 剑指 Offer 27. 二叉树的镜像 简单

题目

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

例如输入：

     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]

题解

1. 递归法，自下而上

递推解析：

a. 当节点root为空，则返回null；

b. 递归，root.left = mirrorTree(root.right)；

c. root.right = mirrorTree(tmp)， tmp用来暂时存储原来的root.left。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode mirrorTree(TreeNode root) {
        if(root == null) return null;
        TreeNode tmp = root.left;
        root.left = mirrorTree(root.right);
        root.right = mirrorTree(tmp); 
        return root;
    }
}

2. 辅助栈

算法流程：
a. 特例处理： 当 root 为空时，直接返回 nullnull ；
b. 初始化： 栈（或队列），本文用栈，并加入根节点 root 。
c. 循环交换： 当栈 stack为空时跳出；
     (1) 出栈： 记为 node；
     (2) 添加子节点： 将 node左和右子节点入栈；
     (3) 交换： 交换 node的左 / 右子节点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode mirrorTree(TreeNode root) {
        if(root == null) return null;
        Stack<TreeNode> st = new Stack<>() {{ add(root); }};
        while(!st.isEmpty()){
            TreeNode node = st.pop();   // 栈顶，也是当前处理的子树的根出栈
            if(node.left != null) st.add(node.left);   // 将非空的左子节点入栈保存，从上到下进行处理
            if(node.right != null) st.add(node.right);  //将非空的右子节点入栈保存
            TreeNode tmp = node.left;  // 交换左右子树
            node.left = node.right;
            node.right = tmp;
        }
        return root;
    }
}