题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
样例
思路分析
构造一个队列容器,即可轻松解决这道题
步骤 | 操作 | 队列 |
1 | 打印节点8 | 节点6、节点10 |
2 | 打印节点6 | 节点10、节点5、节点7 |
3 | 打印节点10 | 节点5、节点7、节点9、节点11 |
4 | 打印节点5 | 节点7、节点9、节点11 |
5 | 打印节点7 | 节点9、节点11 |
6 | 打印节点9 | 节点11 |
7 | 打印节点11 |
每次打印一个节点的时候,如果该节点有子节点,则把该节点的子节点放到一个队列的末尾。接下来到队列的头部取出最早进入队列的节点,重复打印前面的打印操作,直
至队列中所有的节点都被打印出来
此题要注意边界条件,root为空时,不能直接返回null,要返回空的arraylist
代码
方法一:直接利用队列
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
public class Solution{
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<Integer>();
if (root == null){
return list;
}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
while (!queue.isEmpty()){
TreeNode temp = queue.poll(); //根节点出队
if (temp.left != null) {
queue.offer(temp.left);
}
if (temp.right != null) {
queue.offer(temp.right);
}
list.add(temp.val);
}
return list;
}
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
}
结果
方法二:利用arraylist模拟队列操作,(入队:arraylist尾插的特性,出队:remove方法删除0位置节点)
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<Integer>();
if (root == null){
return list;
}
ArrayList<TreeNode> queue = new ArrayList<TreeNode>();
queue.add(root);
while (!queue.isEmpty()){
TreeNode temp = queue.remove(0); //remove方法返回的是删除元素,在这里就是出队操作
if (temp.left != null) {
queue.add(temp.left);
}
if (temp.right != null) {
queue.add(temp.right);
}
list.add(temp.val); //打印操作,每次一个
}
return list;
}
结果
利用arraylist的时间稍快于队列