[2021校招必看之Java版《剑指offer》-18] 二叉树的镜像

1、题目描述

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

  知识点：树、递归
  难度：☆

2、解题思路

2.1 递归思路

  递归最重要的是结束条件，对于本题来说，结束条件就是没有子节点了。
  对于本题，首先假设root的左右子树已经都处理好了，即左子树自身已经镜像了，右子树自身也镜像了，那么最后一步就是交换左右子树，问题解决。所以我只需要将root.left和root.right交换即可。下面进入递归，就是处理子问题。子问题的处理方式与root一样，只是要缩小问题规模，所以只需要考虑root.left是什么情况，root.right是什么情况即可。

2.2 非递归思路

  本问题不用想的那么复杂，本质就是把每一个节点的左子节点和右子节点进行互换，因此，只要遍历二叉树的每一个节点，然后进行子节点互换即可。
  步骤：
    1、判断根节点是否为空，是则返回，否则下一步；
    2、定义一个栈，根节点入栈；
    3、判断栈元素个数是否为0，是则函数结束，否则出栈一个节点node；
    4、交换node的左右子节点；
    5、把node的非空节点入栈；
    6、循环3-5，直到栈元素个数为0，结束函数；

3、解题代码

3.1 递归实现

package pers.klb.jzoffer.simple;

/**
 * @program: JZoffer
 * @description: 二叉树的镜像（递归实现）
 * @author: Meumax
 * @create: 2020-06-17 15:01
 **/
public class ReverseBinaryTree {
    public void Mirror(TreeNode root) {
        if(root==null){
            return;
        }

        TreeNode temp = root.left;
        root.left = root.right;
        root.right = temp;

        Mirror(root.left);
        Mirror(root.right);
    }
}

3.2 非递归实现

package pers.klb.jzoffer.simple;

import java.util.Stack;

/**
 * @program: JZoffer
 * @description: 二叉树的镜像（非递归实现）
 * @author: Meumax
 * @create: 2020-06-17 15:01
 **/
public class ReverseBinaryTree {
    public void Mirror(TreeNode root) {
        // 如果二叉树是空的，就返回
        if (root == null) {
            return;
        }

        // 定义一个栈来保存节点
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(root);
        
        // 循环遍历栈的每一个元素
        while (stack.size() != 0) {
            TreeNode node = stack.pop();

            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;

            if (node.left != null) {
                stack.push(node.left);
            }

            if (node.right != null) {
                stack.push(node.right);
            }
        }
    }
}

  时间复杂度：O(n),n为树节点的个数。每个节点只用遍历一次，所以为O(n)
  空间复杂度：O(n), 每个节点都会在递归栈中存一次

4、解题心得

  递归人人都能想到，这个不难，关键是设置正确的结束条件。
  非递归实现也不难，只要抓住了本题的本质：互换每个节点的左右子节点。然后利用栈、队列等数据结构来进行遍历每一个节点。