辛星Java树算法教程第九篇:二叉树的镜像(递归与非递归)

对于二叉树的镜像，也通常叫做二叉树的反转，这里采用了镜像的说法。
它的操作方法就是把每个节点的左右子树进行交换。
二叉树T和翻转后的二叉树T1会具有一个性质:
T的中序遍历和翻转后T1的中序遍历的顺序正好是相反的。

对于递归形式的二叉树的遍历比较简单，如下:

package com.mengzhidu.teach.algorithm.tree.demo.basic;

import com.mengzhidu.teach.algorithm.tree.demo.TreeNode;
import com.mengzhidu.teach.algorithm.tree.demo.TreeNodeHelper;

/**
 * 用递归的方式来实现二叉树的镜像
 */
public class MirrorDemo1 {

    public static void main(String[] args) {
        TreeNode root = TreeNodeHelper.getTreeNode();
        System.out.println("镜像前的中序遍历是:");
        inOrder(root);
        mirror(root);
        System.out.println("镜像后的前序遍历是:");
        inOrder(root);
    }

    private static void mirror(TreeNode node) {
        if (node == null) {
            return;
        }
        TreeNode tmp = node.getLeft();
        node.setLeft(node.getRight());
        node.setRight(tmp);
        mirror(node.getLeft());
        mirror(node.getRight());
    }

    private static void inOrder(TreeNode node) {
        if (node == null) {
            return;
        }

        inOrder(node.getLeft());
        System.out.println(node.getValue());
        inOrder(node.getRight());
    }
}

对于非递归形式的二叉树，我们可以使用类似于前序遍历的方式来求解，我们可以使用栈这种数据结构来进行处理,其实队列也行，因为先处理哪个子树并不重要，它们之间没有严格的要求，我们这里还是使用了栈，给个范例吧:

package com.mengzhidu.teach.algorithm.tree.demo.basic;

import com.mengzhidu.teach.algorithm.tree.demo.TreeNode;
import com.mengzhidu.teach.algorithm.tree.demo.TreeNodeHelper;

import java.util.Stack;

/**
 * 用非递归的方式来实现二叉树的镜像
 * 这里有点类似于树的前序遍历
 */
public class MirrorDemo2 {

    public static void main(String[] args) {
        TreeNode root = TreeNodeHelper.getTreeNode();
        System.out.println("镜像前的中序遍历:");
        inOrder(root);
        mirrot(root);
        System.out.println("镜像后的中序遍历:");
        inOrder(root);
    }

    private static void mirrot(TreeNode node) {
        Stack<TreeNode> stack = new Stack<>();
        stack.push(node);

        while (!stack.isEmpty()) {
            TreeNode tmp = stack.pop();
            TreeNode left = tmp.getLeft();
            TreeNode right = tmp.getRight();
            tmp.setRight(left);
            tmp.setLeft(right);
            if (left != null) {
                stack.push(left);
            }
            if (right != null) {
                stack.push(right);
            }
        }
    }

    private static void inOrder(TreeNode node) {
        if (node == null) {
            return;
        }
        inOrder(node.getLeft());
        System.out.println(node.getValue());
        inOrder(node.getRight());
    }
}

对于二叉树的镜像的例子，我们就介绍到这里啦。