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

public boolean isComplete(TreeNode root){
if (root == null){
return true;
}
//定义一个isFirstStep变量为true 表示处于第一阶段
boolean isFirstStep = true;
//针对这棵树 进行层序遍历
Queue queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()){
TreeNode cur = queue.poll();
if (isFirstStep){
//第一阶段，要求任意节点必须具备两个子树
if (cur.left != null && cur.right != null){
//当前节点有两个子树，直接把子树用offer方法入队列，继续往后遍历
queue.offer(cur.left);
queue.offer(cur.right);
}else if (cur.left == null && cur.right != null){ //对应笔记中的4号情况
//当前节点左子树为空，右子树非空，这个时候绝对不是完全二叉树 直接return false
return false;
}else if (cur.left != null && cur.right == null){ //对应笔记中的2号情况
//当前节点左子树非空，右子树为空，此时进入第二阶段
//讲isFirstStep置为false，且讲cur.left入队列
isFirstStep = false;
queue.offer(cur.left);
}else {
//当前节点左右子树都为空，进入第二阶段
//对应笔记里面的1号，3号情况
isFirstStep = false;
}
}else{
if (cur.left != null || cur.right != null){ //对应笔记中的5号情况
//当前节点 左子树为空或者右子树为空 那么直接判定不是完全二叉树
return false;
}
}
}
//树遍历完，没有找到反例，最终认定其为完全二叉树
return true;
}