按层遍历,存在两种情况:
(1)当前节点有右孩子却没有左孩子,一定不是完全二叉树
(2)当前节点不是同时有左右两个孩子的时候,后面遍历到的所有节点都必须是叶节点,否则不是完全二叉树
具体方法在注释写的很清楚,代码如下:
public class IsCBT {
public static class Node {
public int value;
public Node left;
public Node right;
public Node(int data) {
this.value = data;
}
}
public static boolean isCBT(Node head) {
if (head == null) {
return true;
}
Queue<Node> queue = new LinkedList<>();
boolean leaf = false; // 是否进入判断叶子节点的阶段
Node l;
Node r;
queue.offer(head);
while (!queue.isEmpty()) {
head = queue.poll(); // 从队列里弹出当前节点
l = head.left;
r = head.right;
if ((leaf && (l != null || r != null)) // 开启了判断叶子节点的阶段,且当前节点不是叶子节点(存在左、右孩子)
||
(l == null && r != null)) { // 左孩子为空,右孩子不为空
return false;
}
// 既然是按层遍历,就应该先加左孩子,再加右孩子
if (l != null) {
queue.offer(l);
}
if (r != null) {
queue.offer(r);
} else { // 左孩子 或 右孩子为空
leaf = true; // 进入判断叶子节点的阶段
}
}
return true;
}
public static void main(String[] args) {
Node head = new Node(4);
head.left = new Node(2);
head.right = new Node(6);
head.left.left = new Node(1);
head.left.right = new Node(3);
head.right.left = new Node(5);
System.out.println(isCBT(head));
}
}