【二叉树】三种方式解决翻转二叉树问题

题目描述

给你一棵二叉树的根节点 root ，翻转这棵二叉树，并返回其根节点。

思路：

可能大家开始看都觉得很懵，但是我们要抓住这道题的本题。所谓的翻转二叉树还不如就叫交换二叉树左右子节点，说到这里是不是就很清晰了。

对于交换二叉树左右子节点来说我们可以采取很多方式，换句话说，如何遍历不重要，只要在遍历过程中，交换遍历到的每个节点的左右子节点就可以了。

方法一： 递归先序遍历（后序遍历也可以）

递归函数的退出条件为当前遍历到的节点为空，每次遍历到一个节点，执行的操作为交换其左右子节点，然后再递归的遍历其左右子树。

    // 翻转二叉树 _ 递归法
    public TreeNode invertTree(TreeNode root) {
        if(root == null){
            return null;
        }
        swap(root);
        invertTree(root.left);
        invertTree(root.right);
        return root;
    }
    
    public void swap(TreeNode root){
        TreeNode temp = root.left;
        root.left = root.right;
        root.right = temp;
    }

方法二：迭代先序遍历

二叉树的递归遍历法可以用迭代的方式写出，因此稍作修改就能得到如下代码。在迭代遍历的过程中交换左右节点即可。

    // 翻转二叉树 _ 迭代法
    public TreeNode invertTree2(TreeNode root){
        if(root == null){
            return root;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while(!stack.isEmpty()){
            TreeNode node = stack.pop();
            if(node.left != null) stack.push(node.left);
            if(node.right != null) stack.push(node.right);
            swap(node);
        }
        return root;
    }

方法三：层序遍历

因为本题只需要交换每个节点的左右节点，因此与层序遍历也是可以解决的，同样是采用在遍历的过程中，交换每个遍历到的节点的左右节点即可。

    // 翻转二叉树 _ 层序遍历
    public TreeNode invertTree3(TreeNode root){
        if(root == null){
            return root;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            for(int i=0; i<size; i++){
                TreeNode node = queue.poll();
                swap(node);
                if(node.left != null) queue.offer(node.left);
                if(node.right != null) queue.offer(node.right);
            }
        }
        return root;
    }

总结：通过这道题的解决，我体会到了二叉树问题的解决离不开二叉树的基础遍历方法，因此，一定要学好并理解基础遍历方法。