反转二叉树

╰半夏微凉° 2021-09-19 16:00 261阅读 0赞

[https://blog.csdn.net/qq\_31692013/article/details/82381671][https_blog.csdn.net_qq_31692013_article_details_82381671]

/**
     * Definition for a binary tree node.
     * public class TreeNode {
     *     int val;
     *     TreeNode left;
     *     TreeNode right;
     *     TreeNode(int x) { val = x; }
     * }
     */
    public class Solution {
        public TreeNode invertTree(TreeNode root) {
            if (root == null) {//一定要检查指针为空
                return null;
            }
            root.left = invertTree(root.left);
            root.right = invertTree(root.right);
     
            TreeNode tmp = root.left;
            root.left = root.right;
            root.right = tmp;
            return root;
        }
    }

反转二叉树

输入一个二叉树，输出其镜像。

如下图，即交换所有节点的左右子树。

![\\][20150616084009267.png]

这里提供两种思路：使用递归和不使用递归。

使用的二叉树定义如下：

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

反转的不同方法：

import java.util.LinkedList;
    
    import java.util.Scanner;
    
     
    
    /*
    
     * 题目描述：输入一个二叉树，输出其镜像。
    
     * */
    
    public class Solution {
        Scanner scanner = new Scanner(System.in);
         
        // 建立二叉树
        public TreeNode createTree(TreeNode root) {
            String val;
            val = scanner.next(); // next方法每次取到一个间隔符前面的数据
            if(val.equals("#")) {
                return null;
            }
            root = new TreeNode(Integer.parseInt(val));  System.out.println("输入的数据为：" + val);
            root.left = createTree(root.left);
            root.right = createTree(root.right);
            return root;
        }
    
        // 得到二叉树的镜像  —— 递归的方式
        public void Mirror(TreeNode root) {
            if(root == null) {
                return;
            }
            if((root.left == null) && (root.right == null)) {
                return;
            }
            TreeNode temp = root.left;
            root.left = root.right;
            root.right = temp;
            Mirror(root.left);
            Mirror(root.right);
        }
    
        // 得到二叉树的镜像 —— 不使用递归
    
        public void MirrorNotRecursive(TreeNode root) {
            java.util.LinkedList stack = new java.util.LinkedList();
            TreeNode temp = null;
            if(root == null) {
                return;
            }
            stack.add(root);
            while(stack.size() != 0) {
                TreeNode node = stack.removeFirst();
                temp = node.left;
                node.left = node.right;
                node.right = temp;
                if(node.right != null) {
                    stack.add(node.right);
                }
                if(node.left != null) {
                    stack.add(node.left);
                }
            }
        }
    
         
    
        // 层次遍历二叉树
    
        public void levelTraverse(TreeNode root) {
            if (root == null) {
                return;
            }
            LinkedList list = new LinkedList();
            list.add(root);
            while (list.size() != 0) {
                TreeNode node = list.removeFirst(); // list.removeFirst() 该方法LinkedList才有
                System.out.print(node.val + " ");
                if(node.left != null) {
                    list.add(node.left);
                }
                if(node.right != null) {
                    list.add(node.right);
                }
            }
        }
       
    
        public static void main(String[] args) {
            Solution solution = new Solution();
            TreeNode root = null;
            root = solution.createTree(root);
            System.out.println("原二叉树的层次遍历");
            solution.levelTraverse(root);
            solution.Mirror(root);
            System.out.println("\n输出该二叉树的镜像");
            solution.levelTraverse(root);
            solution.MirrorNotRecursive(root);
            System.out.println("\n输出该二叉树的镜像（非递归方式）");
            solution.levelTraverse(root);
        }
    }
    
    /*
     * 测试数据：
     * 1 2 3 # 4 # # 5 6 # # # 7 8 # # 9 10 # # 11 # #  （说明：其中#说明左右子树为空）
     * 用先序遍历来建立树后，层次遍历结果为： 1 2 7 3 5 8 9 4 6 10 11
     * 反转二叉树之后：1 7 2 9 8 5 3 11 10 6 4
     */

[https_blog.csdn.net_qq_31692013_article_details_82381671]: https://blog.csdn.net/qq_31692013/article/details/82381671
[20150616084009267.png]: http://img.2cto.com/Collfiles/20150616/20150616084009267.png