题目:
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路:
这道题需要借助队列来实现,队列具有先进先出的特点,先将这棵树的根节点放入队列,只要队列不为空,从队列中出掉最早进入的一个节点,判断这个节点的左子树是不是空,不为空的话入队列,再接着去判断右子树是否为空,不为空再入队列,将出掉的这个节点的值放入一个提前准备好的存储遍历结果的list中,重复前面这些打印步骤,直到队列为空。
画图来说明:
1)下面这棵二叉树中,让根节点8先入队列
2)此时队列不为空了,出掉最早进入的这个元素,也就是8,然后判断8的左子树是否为空,不为空,放入队列,再判断右子树也不为空,入队列,同时将刚刚出掉的这个节点的值放入list中。
3)此时队列也不为空,出掉最早进入队列的节点6,然后分别判断6的左右子树是否为空,不为空,入队列,把6加入list
3)队列不为空,出掉10,10的左右子树都为空,不进行入队列操作,把10加入list,同理,5,7的左右子树也都为空,不入队列,依次出队列,加入list
4)此时队列为空,遍历结束
看代码:
public class CengXuBianLi {
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<>();
if (root == null) {
return null;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
TreeNode treeNode = queue.poll();
if (treeNode.left != null) {
queue.offer(treeNode.left);
}
if (treeNode.right != null) {
queue.offer(treeNode.right);
}
list.add(treeNode.val);
}
return list;
}
}