从上到下打印二叉树就是二叉树的层次遍历,又称为二叉树的广度优先搜索(BFS),通常借助队列实现。
题目一:不分行从上到下打印二叉树
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
提示:
节点总数 <= 1000
实现:
- 创建一个队列,先将根节点入队;
- 队列不为空时,队首元素出队,将该节点的值加入结果集;如果该节点的左(右)孩子不为空,则将左(右)孩子入队,循环打印。
- 队列为空时结束。
参考代码:
class Solution {
public int[] levelOrder(TreeNode root) {
if(root == null)
return new int[0];
ArrayList<Integer> list = new ArrayList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
TreeNode node = queue.poll();
list.add(node.val);
if(node.left != null)
queue.add(node.left);
if(node.right != null)
queue.add(node.right);
}
int[] res = new int[list.size()];
for(int i = 0; i < res.length; i++){
res[i] = list.get(i);
}
return res;
}
}
题目二:分行从上到下打印二叉树
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
提示:
节点总数 <= 1000
思路:
在题目一的基础上,打印每层节点之前求出这一层节点的数目 k,循环 k 次即可打印出这一层的节点。
参考代码:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null)
return new ArrayList<>();
List<List<Integer>> res = new ArrayList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
int k = queue.size();
ArrayList<Integer> temp = new ArrayList<>();
while(k-- != 0){
TreeNode node = queue.poll();
temp.add(node.val);
if(node.left != null)
queue.add(node.left);
if(node.right != null)
queue.add(node.right);
}
res.add(temp);
}
return res;
}
}
题目三:之字形打印二叉树
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
提示:
节点总数 <= 1000
思路一:
存储节点的队列不变,将节点值加入结果集时遵循:奇数层,往队尾添加节点值;偶数层,往队头添加节点值。
参考代码:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null)
return new ArrayList<>();
List<List<Integer>> res = new LinkedList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
queue.add(root);
int level = 0;
while(!queue.isEmpty()){
level++;
int k = queue.size();
LinkedList<Integer> list = new LinkedList<>();
while(k-- != 0){
TreeNode node = queue.poll();
// 添加节点值
if(level % 2 == 1) // 奇数层,addLast
list.addLast(node.val);
else // 偶数层,addFirst
list.addFirst(node.val);
// 节点入队
if(node.left != null)
queue.add(node.left);
if(node.right != null)
queue.add(node.right);
}
res.add(list);
}
return res;
}
}
思路二:
先按照题目二的方法分行打印二叉树,最后利用Collections.reverse()
方法对结果集中偶数层的list
进行反转。
参考代码:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null)
return new ArrayList<>();
List<List<Integer>> res = new ArrayList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
queue.add(root);
int level = 0;
while(!queue.isEmpty()){
level++;
int k = queue.size();
ArrayList<Integer> temp = new ArrayList<>();
while(k-- != 0){
TreeNode node = queue.poll();
temp.add(node.val);
if(node.left != null)
queue.add(node.left);
if(node.right != null)
queue.add(node.right);
}
// 偶数层,反转
if(level % 2 == 0)
Collections.reverse(temp);
res.add(temp);
}
return res;
}
}