LeetCode-226-翻转二叉树

题目:

给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。
思路:对于叶子节点,不做考虑,直接continue。对于非叶子节点,如果其有两个孩子节点,将这两个孩子节点互换。之后将这两个孩子节点加入队列,以便后续考虑。如果只有一个孩子节点,假设有左孩子节点,将其作为右孩子节点,原左孩子节点设为null。
第一个continue是为了不在对叶子节点做多余判断。
第二个continue,是因为我们将node的左、右孩子节点已经加入了队列,不在执行下面代码,防止重复加入队列。
第三个continue,判断条件为node有左孩子,我们可知,其必没有右孩子节点,正常思路应该不执行

if(node.right != null){
	node.left = node.right;
	node.right = null;
	queue.add(node.left);
}

但是我们对换了左孩子和右孩子,此时右孩子不为null,所以此时直接跳过,以防执行下面的语句。
完整代码如下:


class Solution {
    public TreeNode invertTree(TreeNode root) {
        if(root == null){
            return null;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        TreeNode node = null;
        queue.add(root);
        while(!queue.isEmpty()){
            node = queue.poll();
            if(node.left == null && node.right == null){
                continue;
            }
            if(node.left != null && node.right != null){
                TreeNode temp = node.left;
                node.left = node.right;
                node.right = temp;
                queue.add(node.left);
                queue.add(node.right);
                continue;
            }
            //root.right必是null
            if(node.left != null){
                node.right = node.left;
                node.left = null;
                queue.add(node.right);
                continue;
            }
            if(node.right != null){
                node.left = node.right;
                node.right = null;
                queue.add(node.left);
            }
        }
        return root;
    }
}
  • 6
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值