题目地址:
https://leetcode.com/problems/check-completeness-of-a-binary-tree/
判断一棵二叉树是否是complete的。可以用BFS。
法1:BFS(不允许offer null节点)。每offer进一个节点的时候,做以下操作步骤:
1、如果该节点左孩子为null而右孩子非null,直接返回false;
2、如果左孩子或右孩子有一个为null,就做一下标记。在此之后入队的任何节点,只要它有孩子,就返回false;
3、把该节点的孩子offer进队列。
代码如下:
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
public boolean isCompleteTree(TreeNode root) {
if (root == null) {
return true;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
// 如果遇到某个节点只有右孩子而没有左孩子,就做下标记
// 接下来入队的节点都不能有孩子
boolean alert = false;
while (!queue.isEmpty()) {
TreeNode x = queue.poll();
// 如果在已经做过标记的情况下,当前节点有孩子,就直接返回false
if (alert && (x.left != null || x.right != null)) {
return false;
}
// 如果当前节点只有右孩子没左孩子,也返回false
if (x.left == null && x.right != null) {
return false;
}
// 如果当前节点孩子缺一个,就做一下标记
if (x.left == null || x.right == null) {
alert = true;
}
// 接下来把孩子入队
if (x.left != null) {
queue.offer(x.left);
}
if (x.right != null) {
queue.offer(x.right);
}
}
return true;
}
}
class TreeNode {
int val;
TreeNode left, right;
TreeNode(int x) {
val = x;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( n ) O(n) O(n)。
法2:BFS(允许offer null节点,注意一定要用LinkedList,ArrayDeque不允许offer null进去)。在遍历的时候一旦遇到null节点,说明二叉树已经遍历结束了,接下来队列里的节点必须全是null,否则就返回false。
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
public boolean isCompleteTree(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
boolean end = false;
while (!queue.isEmpty()) {
TreeNode x = queue.poll();
// 如果已经到末尾了,队列里还有非null元素,返回false
if (end && x != null) {
return false;
}
// 如果当前节点是null,说明已经遍历到末尾了
if (x == null) {
end = true;
} else {
// 否则把左右孩子都入队
queue.offer(x.left);
queue.offer(x.right);
}
}
return true;
}
}
时空复杂度 O ( n ) O(n) O(n)。可以这么想,一个node对应两个children,复杂度量级仍然和树的节点数是相同的。