BM33-二叉树的镜像

题目

操作给定的二叉树，将其变换为源二叉树的镜像。

数据范围：二叉树的节点数 0≤n≤1000， 二叉树每个节点的值 0≤val≤1000。

要求： 空间复杂度 O(n)。本题也有原地操作，即空间复杂度 O(1)的解法，时间复杂度 O(n)。

比如：

源二叉树

镜像二叉树

示例1

输入：{8,6,10,5,7,9,11}

返回值：{8,10,6,11,9,7,5}

示例2

输入：{}

返回值：{}

---

思路1：递归

先递归处理二叉树的左子树，再递归处理二叉树的右子树，然后将原二叉树根节点的左子树连接处理后的右子树，将原二叉树根节点的右子树连接处理后的左子树，即可完成二叉树的镜像。

---

代码1

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return TreeNode类
     */
    public TreeNode Mirror (TreeNode pRoot) {
        // write code here
        if(pRoot == null || (pRoot.left == null && pRoot.right == null)) {
            return pRoot;
        }

        TreeNode left = Mirror(pRoot.left);
        TreeNode right = Mirror(pRoot.right);

        pRoot.left = right;
        pRoot.right = left;

        return pRoot;
    }
}

---

思路2：栈

二叉树中能够用递归的，大多也可以用栈来实现。栈的访问是一种自顶向下的访问，因此需要在左右子节点入栈后直接交换，然后再访问后续栈中内容。

具体做法：

• step 1：优先检查空树的情况。
• step 2：使用栈辅助遍历二叉树，根节点先进栈。
• step 3：遍历过程中每次弹出栈中一个元素，然后该节点左右节点分别入栈。
• step 4：同时我们交换入栈两个子节点的值，因为子节点已经入栈了再交换，就不怕后续没有交换。

---

代码2

import java.util.*;

public class Solution {
    public TreeNode Mirror (TreeNode pRoot) {
        //空树
        if(pRoot == null) {
            return null;
        }

        //辅助栈
        Stack<TreeNode> s = new Stack<>();
        //根节点先进栈
        s.push(pRoot);

        while(!s.isEmpty()) {
            TreeNode node = s.pop();
            //左右节点入栈
            if(node.left != null) {
                s.push(node.left);
            }
            if(node.right != null) {
                s.push(node.right);
            }

            //交换左右
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;
        }
        return pRoot;
    }
}