【leetcode】【Easy】【226. Invert Binary Tree】【tree】

最新推荐文章于 2020-12-28 21:55:47 发布

原创最新推荐文章于 2020-12-28 21:55:47 发布 · 472 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

LeetCode_Easy 专栏收录该内容

13 篇文章

订阅专栏

本文介绍了两种实现二叉树节点翻转的方法：递归方式和广度优先搜索（BFS）。递归方法通过交换每个节点的左右子树来翻转整棵树，在LeetCode上的性能优于BFS。BFS则通过队列遍历并交换每个节点的子树。

problem link

code：

code1:递归在leetcode上要比BFS快一点

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

code2：BFS遍历整个树，比递归慢

/**
 * 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;
            
        Queue<TreeNode> queue=new LinkedList<TreeNode>();
        queue.offer(root);
        
        while(!queue.isEmpty()){
            TreeNode cur=queue.poll();
            
            //要考虑左右子树是否存在
            if(cur.left==null && cur.right!=null){//只存在右子树
                cur.left=cur.right;
                cur.right=null;
                queue.offer(cur.left);
            }else if(cur.right==null && cur.left!=null){//只存在左子树
                cur.right=cur.left;
                cur.left=null;
                queue.offer(cur.right);
            }else if(cur.left!=null && cur.right!=null){//左右子树都存在
                TreeNode temp=cur.left;
                cur.left=cur.right;
                cur.right=temp;
            
                queue.offer(cur.left);
                queue.offer(cur.right);
            }//左右子树都不存在不用操作
            
        }
        
        return root;
    }
}

while循环中的判断可以简化。

但是不懂为什么下面三行不会报NullPointerException？？

TreeNode temp=cur.left;
cur.left=cur.right;
cur.right=temp;

/**
 * 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;
            
        Queue<TreeNode> queue=new LinkedList<TreeNode>();
        queue.offer(root);
        
        while(!queue.isEmpty()){
            TreeNode cur=queue.poll();
            
            TreeNode temp=cur.left;
            cur.left=cur.right;
            cur.right=temp;
            
            if(cur.left!=null){
                queue.offer(cur.left);
            }
            
            if(cur.right!=null){
                queue.offer(cur.right);
            }
        }
        
        return root;
    }
}