剑指 Offer 32 - II. 从上到下打印二叉树 II

难度简单64
从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

  1.     3
  2.    / \
  3.   9  20
  4.     /  \
  5.    15   7

返回其层次遍历结果：

  1. [
  2.   [3],
  3.   [9,20],
  4.   [15,7]
  5. ]

提示：

  1. 节点总数 <= 1000

    Solution

    1. /**
    2. * Definition for a binary tree node.
    3. * public class TreeNode {
    4. *     int val;
    5. *     TreeNode left;
    6. *     TreeNode right;
    7. *     TreeNode(int x) { val = x; }
    8. * }
    9. */
    10. class Solution {
    11.  public List<List<Integer>> levelOrder(TreeNode root) {
    13.      Queue<TreeNode> queue = new LinkedList<>();
    14.      List<List<Integer>> res = new ArrayList<>();
    15.      int cur = 0;  // 当前层次元素的个数
    16.      int next = 0;  // 统计下一层元素的个数
    17.      if ( root != null) {
    18.          queue.offer(root);
    19.          cur ++;
    20.      }
    21.      List<Integer> temp = new ArrayList<>();
    22.      while (!queue.isEmpty()){
    23.          TreeNode node = queue.poll();
    24.          temp.add(node.val);
    25.          cur --;
    27.          if (node.left != null){
    28.              queue.offer(node.left);
    29.              next ++;
    30.          } 
    31.          if (node.right != null){
    32.              queue.offer(node.right);
    33.              next ++;
    34.          }
    35.          if (cur == 0){
    36.              res.add(temp);
    37.              temp = new ArrayList<>();
    38.              cur = next;
    39.              next = 0;
    40.          }
    41.      }
    42.      return res; 
    43.  }
    44. }