判断一棵树是不是完全二叉树
- 对二叉树进行层序遍历
- 遍历过程中,状态分成两个阶段
a)第一阶段,每个节点都有两个子树
b)第二阶段,遇到一个特殊的节点,可能只有左子树可能没有子树,就开启第二阶段(这个临界节点如果是有右子树,没有左子树,那么说明一定不是完全二叉树)
c)第二阶段中,接下来每个节点都必须没有子树,如果要是有子树,就不是完全二叉树
public boolean isComplete(TreeNode root){
if(root == null){
return true;
}
//层序遍历
//这是个标记,如果这个标记为true,接下来的节点不能有子树,即第二阶段开始
boolean needNoChild = false;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()){
TreeNode cur = queue.poll();
//对这个元素情况进行判定,访问是一组比较复杂的判断
//第一阶段,这个节点必须有两个子树
if(!needNoChild){
//第一阶段的情况
if(cur.left != null && cur.right != null){
//合格的状态,左右子树加入队列
queue.offer(cur.left);
queue.offer(cur.right);
}else if (cur.left == null && cur.right != null){
return false;
}else if(cur.left != null && cur.right == null){
queue.offer(cur.left);
needNoChild = true;
}else {
needNoChild = true;
}
}else {
//第二阶段情况
if(cur.left != null || cur.right != null){
return false;
}
}
}
return true;
}
思路:
完全二叉树,层序遍历后,都是连续的,不会出现中间是空的情况
如果发现某个节点是空,后面的节点也必须为空
public boolean isComleteTree(TreeNode root){
if(root == null){
return true;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
boolean flag = false;
while (!queue.isEmpty()){
TreeNode curNode = queue.poll();
if(curNode == null){
//flag 用来开启状态,如果遇见取出的元素为空,跳过该层循环进入下一层循环
flag = true;
continue;
}
if(flag){
return false;
}
queue.offer(curNode.left);
queue.offer(curNode.right);
}
return true;
}