按層遍歷,存在兩種情況:
(1)當(dāng)前節(jié)點(diǎn)有右孩子卻沒(méi)有左孩子,一定不是完全二叉樹(shù)
(2)當(dāng)前節(jié)點(diǎn)不是同時(shí)有左右兩個(gè)孩子的時(shí)候,后面遍歷到的所有節(jié)點(diǎn)都必須是葉節(jié)點(diǎn),否則不是完全二叉樹(shù)
具體方法在注釋寫(xiě)的很清楚,代碼如下:
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; // 是否進(jìn)入判斷葉子節(jié)點(diǎn)的階段
Node l;
Node r;
queue.offer(head);
while (!queue.isEmpty()) {
head = queue.poll(); // 從隊(duì)列里彈出當(dāng)前節(jié)點(diǎn)
l = head.left;
r = head.right;
if ((leaf && (l != null || r != null)) // 開(kāi)啟了判斷葉子節(jié)點(diǎn)的階段,且當(dāng)前節(jié)點(diǎn)不是葉子節(jié)點(diǎn)(存在左、右孩子)
||
(l == null && r != null)) { // 左孩子為空,右孩子不為空
return false;
}
// 既然是按層遍歷,就應(yīng)該先加左孩子,再加右孩子
if (l != null) {
queue.offer(l);
}
if (r != null) {
queue.offer(r);
} else { // 左孩子 或 右孩子為空
leaf = true; // 進(jìn)入判斷葉子節(jié)點(diǎn)的階段
}
}
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));
}
}