【Leetcode】958. Check Completeness of a Binary Tree

题目地址:

https://leetcode.com/problems/check-completeness-of-a-binary-tree/

判断一棵二叉树是否是complete的。可以用BFS。

法1:BFS(不允许offer null节点)。每offer进一个节点的时候,做以下操作步骤:
1、如果该节点左孩子为null而右孩子非null,直接返回false;
2、如果左孩子或右孩子有一个为null,就做一下标记。在此之后入队的任何节点,只要它有孩子,就返回false;
3、把该节点的孩子offer进队列。
代码如下:

import java.util.LinkedList;
import java.util.Queue;

public class Solution {
    public boolean isCompleteTree(TreeNode root) {
        if (root == null) {
            return true;
        }
        
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        // 如果遇到某个节点只有右孩子而没有左孩子,就做下标记
        // 接下来入队的节点都不能有孩子
        boolean alert = false;
        while (!queue.isEmpty()) {
            TreeNode x = queue.poll();
            // 如果在已经做过标记的情况下,当前节点有孩子,就直接返回false
            if (alert && (x.left != null || x.right != null)) {
                return false;
            }
            // 如果当前节点只有右孩子没左孩子,也返回false
            if (x.left == null && x.right != null) {
                return false;
            }
            // 如果当前节点孩子缺一个,就做一下标记
            if (x.left == null || x.right == null) {
                alert = true;
            }
            // 接下来把孩子入队
            if (x.left != null) {
                queue.offer(x.left);
            }
            if (x.right != null) {
                queue.offer(x.right);
            }
        }
        
        return true;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    TreeNode(int x) {
        val = x;
    }
}

时间复杂度 O ( n ) O(n) O(n),空间 O ( n ) O(n) O(n)

法2:BFS(允许offer null节点,注意一定要用LinkedList,ArrayDeque不允许offer null进去)。在遍历的时候一旦遇到null节点,说明二叉树已经遍历结束了,接下来队列里的节点必须全是null,否则就返回false。

import java.util.LinkedList;
import java.util.Queue;

public class Solution {
    public boolean isCompleteTree(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        boolean end = false;
        while (!queue.isEmpty()) {
            TreeNode x = queue.poll();
            // 如果已经到末尾了,队列里还有非null元素,返回false
            if (end && x != null) {
                return false;
            }
            // 如果当前节点是null,说明已经遍历到末尾了
            if (x == null) {
                end = true;
            } else {
            	// 否则把左右孩子都入队
                queue.offer(x.left);
                queue.offer(x.right);
            }
        }
        
        return true;
    }
}

时空复杂度 O ( n ) O(n) O(n)。可以这么想,一个node对应两个children,复杂度量级仍然和树的节点数是相同的。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值