剑指Offer - 32-2_从上到下打印二叉树

题目描述

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

示例

输入：
	3
   / \
  9  20
    /  \
   15   7
输出：
[
  [3],
  [9,20],
  [15,7]
]

思路

1. 采用广度优先搜索，从头结点开始遍历，先把头结点添加进队列
2. 然后从队列取结点，取的数量为当前队列的大小，这些刚好都是同一层的节点
3. 然后依次把值添加进对应层的结果集中，然后判断当前节点有没有左右结点，如果有，依次添加进队列，然后把当前层的结果集添加进整体结果集中
4. 以此类推，遍历队列，最后得到结果

Code

public class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        if (null == root) {
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
            //从队列中去当前队列大小个结点
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                list.add(node.val);
                if (null != node.left) {
                    queue.add(node.left);
                }
                if (null != node.right) {
                    queue.add(node.right);
                }
            }
            result.add(list);
        }
        return result;
    }
}