BFS(Breadth-First Search)是常见算法之一广度优先遍历,即按层及顺序遍历。
思路分析
- 借助队列先进先出的特性,将当前一行的节点以此推入队列
- 在弹出这一行的每个节点的同时,将弹出节点的子节点推入队列(子节点属于下一行),从而实现广度优先
模板如下
public List<Integer> BFS(TreeNode root) {
//定义一个数组接受每个节点的值
List<Integer> res = new ArrayList<>();
//很关键,不判断的话root为空时下面queue.peek()会报错
if(root == null){
return res;
}
//定义一个存放树节点的队列
Queue<TreeNode> queue = new LinkedList<>();
//现将根节点推入队列,此时队列只有一个节点
queue.offer(root);
while(!queue.isEmpty()) {
//在这获取队列的长度,为下面for遍历这一层的节点做准备
int size = queue.size();
res.add(queue.peek().val);
//遍历 (n=当前队列长度) 次,拿出所有属于这一层的节点
for(int i = 0; i < size; i++){
//用cur接收队列弹出的TreeNode,此时队列长度-1
TreeNode cur = queue.p