1月3日代码随想录反转二叉树-Toy模板网

这篇具有很好参考价值的文章主要介绍了1月3日代码随想录反转二叉树。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

226翻转二叉树

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

示例 1：

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

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

示例 2：

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

输入：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;
    }