首先是广度优先遍历,借助于一个队列来实现,首先把root放到队列当中,之后进入循环,拿出第一个元素,访问,依次把它的左右孩子放到队列当中(如果有的话),继续循环。
public void bfs(TreeNode root){
if (root != null){
Queue<TreeNode> queue = new LinkedList<TreeNode>();
TreeNode node = root;
queue.offer(node);
while(!queue.isEmpty()){
node = queue.poll();
node.visit();
if (node.getLeft() != null){
queue.offer(node.getLeft());
}
if (node.getRight() != null){
queue.offer(node.getRight());
}
}
}
}
关于完全二叉树的判定,这里是以广度优先便利为基础,引入了一个变量,step,它的意义是这样的,已遍历过得所有节点,度都等于2,那么step = 2;如果出现了一个节点,只有一个左孩子,step=1,需要注意的是,这样的节点,最多有一个,如果有的话,它的后续节点,都必须是叶子节点;从第一个叶子节点开始,step = 0;
另外引入了rightExist、leftExist、exist三个变量,exist有4个可能的值,0,1,2,3,代表左右节点的存在情况。
下面重点分析一下几种情况:
exist=0,单纯的把step=0即可
exist=1,只有右孩子,这种情况直接返回false
exist=2,只有左孩子,需要判定,step=2,也就是说,前面所有节点的度都等于2,继续遍历,同时把step=1;否则,返回false
exit=3,左右都有,需要判定,step=2,也就是说,前面所有节点的度都等于2,继续遍历;否则,返回false
public boolean isComplete(TreeNode root){
if (root != null){
Queue<TreeNode> queue = new LinkedList<TreeNode>();
TreeNode node = root;
queue.offer(node);
int step = 2;
while(!queue.isEmpty()){
node = queue.poll();
int rightExist = node.getRight() == null ? 0 : 1;
int leftExist = node.getLeft() == null ? 0 : 1;
int exist = rightExist | (leftExist << 1);
switch(exist){
case 0:
step = 0;
break;
case 1:
return false;
case 2:
if (step == 2){
step = 1;
queue.offer(node.getLeft());
break;
}else{
return false;
}
case 3:
if (step == 2){
queue.offer(node.getLeft());
queue.offer(node.getRight());
break;
}else{
return false;
}
default:
break;
}
}
}
return true;
}