1月3日代码随想录反转二叉树

226翻转二叉树

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

示例 1：

输入：root = [4,2,7,1,3,6,9]
输出：[4,7,2,9,6,3,1]

示例 2：

输入：root = [2,1,3]
输出：[2,3,1]

示例 3：

输入：root = []
输出：[]

提示：

• 树中节点数目范围在?[0, 100]?内
• -100 <= Node.val <= 100

思路

这道题用递归的思想就是将根节点的左右儿子交换，然后再对子节点进行递归操作，直到子节点均为空。

但是我感觉那个TreeNode类型的函数拿来做递归有点奇怪，所以另外定义了一个递归函数。

class Solution {
    public TreeNode invertTree(TreeNode root) {
        if(root==null)
            return null;
        invert(root);
        return root;

    }
    public void invert(TreeNode node){
        if(node.left!=null||node.right!=null){
            TreeNode tmp=node.left;
            node.left=node.right;
            node.right=tmp;
            if(node.left!=null) invert(node.left);
            if(node.right!=null) invert(node.right);
        }
    }
}

然后看了一下题解的递归，只用了一个函数就行了，只要把递归的结果作为交换过的节点值存储起来即可。

class Solution {
    public TreeNode invertTree(TreeNode root) {
        if (root == null) {
            return null;
        }
        TreeNode left = invertTree(root.left);
        TreeNode right = invertTree(root.right);
        root.left = right;
        root.right = left;
        return root;
    }
}

还有一种方法是层序遍历的同时将子节点交换。

 public TreeNode invertTreeByQueue(TreeNode root) {
        if (root == null) {
            return null;
        }
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;
            if (node.left != null) {
                queue.offer(node.left);
            }
            if (node.right != null) {
                queue.offer(node.right);
            }
        }
        return root;
    }

总结

这些方法实际上都使用了二叉树的前序、后序遍历法，可以用递归，层序遍历等方法实现。