剑指Offer-从上往下分行打印二叉树

25.从上往下分行打印二叉树

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

样例

输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
    8
   / \
  12  2
     /
    6
   /
  4

输出：[[8], [12, 2], [6], [4]]

思路

与不分行打印思路差不多,只是每层需要用一个list缓存一下这行的元素，每处理完一行，如果缓存列表中有元素，要加入到结果集中。

class Solution {
    public List<List<Integer>> printFromTopToBottom(TreeNode root) {
        List<List<Integer>> res=new ArrayList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        if(root==null) return res;
        
        queue.add(root);
        while(!queue.isEmpty())
        {
            List<Integer> mres=new  ArrayList<>();
            int size=queue.size();
            while(size-->0)
            {
                TreeNode temp=queue.poll();
                if(temp==null) continue;
                mres.add(temp.val);
                queue.add(temp.left);
                queue.add(temp.right);
            }
            if(mres.size()!=0) res.add(mres);
        }
        return res;
    }
}