目录
- 题目思路
- BFS
题目来源
剑指 Offer 32 - II. 从上到下打印二叉树 II
题目思路
- I. 按层打印: 题目要求的二叉树的 从上至下 打印(即按层打印),又称为二叉树的 广度优先搜索(BFS)。BFS 通常借助 队列 的先入先出特性来实现。
- II. 每层打印到一行: 将本层全部节点打印到一行,并将下一层全部节点加入队列,以此类推,即可分为多行打印。
BFS
- 特例处理: 当根节点为空,则返回空列表 [] ;
- 初始化: 打印结果列表 result = [] ,包含根节点的队列 queue.add(root) ;
- 新建一个临时列表 list,用于存储当前层打印结果;
- BFS 循环: 当队列 queue 为空时跳出;
- 当前层打印循环: 循环次数为当前层节点数(即队列 queue 长度);
- 出队: 队首元素出队,记为 node;
- 打印: 将 node.val 添加至 list尾部;
- 添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ;
- 将当前层结果 list添加入 result。
- 删除list的元素
- 返回值: 返回打印结果列表 result 即可。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
List<Integer> list = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(root != null) queue.add(root);
while(!queue.isEmpty()){
for(int i = queue.size();i>0;i--){
TreeNode node = queue.poll();
list.add(node.val);
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
}
result.add(new ArrayList(list));
list.clear();
}
return result;
}
}