【剑指 Offer 27】二叉树的镜像

news2025/1/15 22:34:12

题目：

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

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

输入输出样例
在这里插入图片描述

思考1：

二叉树的镜像，就是交换二叉树的每个节点的左右结点
所以想到递归
root 为空代表这棵子树是一个空树了，直接返回 null；root 的左右结点都为空，代表这棵子树只有根结点了，不需要交换直接返回
交换每个根结点的左右子结点

题解：

class Solution {

    public TreeNode mirrorTree(TreeNode root) {
        if(root == null) return null;

        if (root.left == null && root.right == null) return root;

        mirrorTree(root.left);
        mirrorTree(root.right);

        TreeNode tmp = root.left;
        root.left = root.right;
        root.right = tmp;

        return root;
    }
}

思考2：

也可以借助辅助栈
先 new 一个辅助栈，将头结点入栈，然后开始循环
先出栈 node，如果 node 的左右子结点哪个不为空，就入栈，然后交换 node 的左右子结点
直到栈空停止循环
栈其实就是遍历所有节点，然后交换左右子节点

题解：

class Solution {
    public TreeNode mirrorTree(TreeNode root) {
        if (root == null) return null;

        Stack<TreeNode> stack = new Stack<>();
        stack.add(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 tmp = node.left;
            node.left = node.right;
            node.right = tmp;
        }
        return root;
    }
}