二叉树结构

  1. class TreeNode {
  2.     int val = 0;
  3.     TreeNode left = null;
  4.     TreeNode right = null;
  6.     public TreeNode(int val) {
  7.         this.val = val;
  8.     }
  9. }

题目描述

  • 从上往下打印出二叉树的每个节点，同层节点从左至右打印。

解析

  • 考察层序遍历
  • 每一次打印一个结点的时候，如果该结点有子节点，则把该结点的子节点放到一个队列的末尾。
  • 接下来到队列的头部取出最早进入队列的结点，重复前面的打印操作，直至队列中所有的结点都被打印出来为止。

代码实现

  1. public ArrayList<Integer> printFromTopToBottom(TreeNode root) {
  2.     ArrayList<Integer> list = new ArrayList<Integer>();
  3.     if(root==null){
  4.         return list;
  5.     }
  6.     Queue<TreeNode> queue = new LinkedList<TreeNode>();
  7.     queue.offer(root);
  8.     while (!queue.isEmpty()) {
  9.         TreeNode treeNode = queue.poll();
  10.         if (treeNode.left != null) {
  11.             queue.offer(treeNode.left);
  12.         }
  13.         if (treeNode.right != null) {
  14.             queue.offer(treeNode.right);
  15.         }
  16.         list.add(treeNode.val);
  17.     }
  18.     return list;
  19. }

辅助包

  1. import java.util.ArrayList;
  2. import java.util.LinkedList;
  3. import java.util.Queue;