【剑指offer】队列+BFS+二叉树三道典型例题
简单介绍一下BFS
其过程简单来说是对每一层节点依次访问,访问完一层进入下一层,而且每个节点只能访问一次。
对于上面的例子,BFS的结果就是A,B,C,D,E,F,G,H,I(假设每层节点从左到右访问)。
广度优先遍历各个节点,需要使用队列(Queue)这种数据结构,其特点是先进先出。也可以使用双端队列,区别就是双端队列首尾都可以插入和弹出节点。整个遍历过程如下:
1)首先将A节点插入队列中,queue(A);
2)将A节点弹出,同时将A的子结点B、C插入队列中,此时B在队列首部,C在队列尾部,queue(B,C);
3)将B节点弹出,同时将B的子结点D、E插入队列中,此时C在队列首部,E在队列尾部,queue(C,D,E);
4)将C节点弹出,同时将C的子节点F,G,H插入队列中,此时D在队列首部,H在队列尾部,queue(D,E,F,G,H);
5)将D节点弹出,D没有子节点,此时E在队列首部,H在队列尾部,queue(E,F,G,H);
…依次往下,最终遍历完成。
BFS模板代码
void bfs(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
}
由题意知,需要按层遍历,得到最后一层最左侧节点的值,所以可以利用BFS来层序遍历这棵二叉树。不过,仅仅利用BFS不能判断哪个节点才是最左侧的节点,所以要加一些判断。要遍历某一层时,获取此时队列的size,此时size的大小即为该层节点的个数,然后利用for循环限制遍历次数,第一次遍历到的节点即为该层最左侧节点。
public int findBottomLeftValue(TreeNode root) {
Queue<TreeNode> q=new LinkedList<>();
int ret=0;
q.offer(root);
while(q.size()!=0){
int size=q.size();
for(int i=0;i<size;i++){
TreeNode t=q.poll();
if(i==0) ret=t.val;
if(t.left!=null) q.offer(t.left);
if(t.right!=null) q.offer(t.right);
}
}
return ret;
}
整体思路和上一题几乎相等,只是在外层循环定义一个变量用于更新每一层的最大值;
class Solution {
public List<Integer> largestValues(TreeNode root) {
Queue<TreeNode> q=new LinkedList<>();
List<Integer> list=new ArrayList<>();
if(root==null) return list;
q.offer(root);
while(q.size()!=0){
int size=q.size();
int max=Integer.MIN_VALUE;
for(int i=0;i<size;i++){
TreeNode t=q.poll();
if(max<t.val) max=t.val;
if(t.left!=null) q.offer(t.left);
if(t.right!=null) q.offer(t.right);
}
list.add(max);
}
return list;
}
}
同样使用BFS层序遍历这课二叉树,当遍历到每层最后一个节点是,把这个节点加入List;
class Solution {
public List<Integer> rightSideView(TreeNode root) {
Queue<TreeNode> q=new LinkedList<>();
List<Integer> list=new ArrayList<>();
if(root==null) return list;
q.offer(root);
while(q.size()!=0){
int size=q.size();
for(int i=0;i<size;i++){
TreeNode t=q.poll();
if(i==size-1){
list.add(t.val);
}
if(t.left!=null) q.offer(t.left);
if(t.right!=null) q.offer(t.right);
}
}
return list;
}
}
BFS在解决层序遍历时非常有用,对于BFS的实现过程要多加理解,尤其理解为什么要使用队列来辅助进行BFS;
|