剑指 Offer 27. 二叉树的镜像

剑指 Offer 27. 二叉树的镜像

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

递归

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;
    }
}

栈

辅助栈（或队列）
利用栈（或队列）遍历树的所有节点 nodenode ，并交换每个 nodenode 的左 / 右子节点。
算法流程：
特例处理： 当 rootroot 为空时，直接返回 nullnull ；
初始化： 栈（或队列），本文用栈，并加入根节点 rootroot 。
循环交换： 当栈 stackstack 为空时跳出；
出栈： 记为 nodenode ；
添加子节点： 将 nodenode 左和右子节点入栈；
交换： 交换 nodenode 的左 / 右子节点。
返回值： 返回根节点 rootroot 。

class Solution {
    public TreeNode mirrorTree(TreeNode root) {
        if(root == null) return null;
        Stack<TreeNode> stack = new 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;
    }
}