【剑指Offer】32 - ll. 从上到下打印二叉树 ll

题目描述

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

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

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

解题思路

I：按层打印：题目要求的二叉树从上到下打印（即按层次打印），又称为二叉树的广度优先搜索（BFS）。BFS通常借助队列的的先入先出特性。

II：每层打印到一行： 将本层全部节点打印到一行，并将下一层全部节点加入队列，以此类推，即可分为多行打印。

代码实现

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        ArrayList<List<Integer>> arrayList = new ArrayList<>();
        ArrayDeque<TreeNode> deque = new ArrayDeque<>();
        if(null != root)
            deque.add(root);
        while (!deque.isEmpty()) {
        	// 创建一个临时集合装入每一层元素的值
            List<Integer> list = new ArrayList<>();
            for (int i = deque.size(); i > 0; i--) {
                TreeNode node = deque.poll();
                list.add(node.val);
                if (node.left != null)
                    deque.add(node.left);
                if (node.right != null)
                    deque.add(node.right);
            }
            arrayList.add(list);
        }
        return arrayList;
    }
}