判断一个树是否为完全二叉树

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

class TreeNode {
    public int val;
    public TreeNode left;
    public TreeNode right;

    public TreeNode(int val) {
        this.val = val;
    }
}

public class CompleteTree {
    // 对于完全二叉树, 在层序遍历时, 会出现两种状态:
    // 1. 任何一个节点都有左右子树
    //    当遇到某个节点没有子树或只有左子树, 切换到第二状态
    //    如果遇到某个节点, 只有右子树, 那他一定不是完全二叉树
    // 2. 任何一个节点, 一定没有子树

    public boolean isCompleteTree(TreeNode root) {
        // 创建一个标志位, 判断是第一阶段还是第二阶段
        boolean isSecondStep = false;

        if (root == null) {
            return true;
        }

        // 针对这个树进行层序遍历
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            TreeNode cur = queue.poll();
            // 访问这个节点(判断是否符合完全二叉树的要求)
            if (!isSecondStep) {
                // 第一阶段
                if (cur.left != null && cur.right != null) {
                    queue.offer(root.left);
                    queue.offer(root.right);
                } else if (cur.right != null && cur.left == null) {
                    // 只有右子树, 没有左子树, 一定不是完全二叉树
                    return false;
                } else if (cur.left != null && cur.right == null) {
                    // 只有左子树, 没有右子树,切换到第二形态
                    isSecondStep = true;
                } else {
                    // 第二阶段
                    // 第二阶段任何一个节点都没有子树
                    if (cur.left != null || cur.right != null) {
                        return false;
                    }
                }
            }
        }
        // 循环结束, 整个树都遍历完了, 还是没有找到反例, 说明这个数是完全二叉树
        return true;
    }

}

上面这种写法不能通过 leetcode , 超过时间限制; 改进后如下:
直接用当前节点进行判断; 还是将二叉树按照层序遍历的方法依次入队列, 如果当前为空, 此时有两种情况, 如果还会遇到不为 null 的节点, 说明这个树不是完全二叉树, 直接返回 false , 第二种情况就是后面一直是 null , 知道==直到循环结束,此时这个树就是完全二叉树, 返回 true;

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

class TreeNode{
    int val;
    TreeNode left;
    TreeNode right;

    public TreeNode(int val) {
        this.val = val;
    }
}

public class isCompleteTree {
    public boolean isCompleteTree(TreeNode root) {
        if (root == null) {
            // 如果二叉树是空树, 认为他是完全二叉树
            return true;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        boolean flag = false;
        while (!queue.isEmpty()) {
            TreeNode cur = queue.poll();
            if (cur == null) {
                flag = true;
                continue;
            }
            if (flag) {
                // 此时说明空子树后面还有子树
                return false;
            }
            queue.offer(cur.left);
            queue.offer(cur.right);
        }
        return true;
    }
}