剑指 Offer 32 - II. 从上到下打印二叉树 II
从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7
返回其层次遍历结果：
[ [3], [9,20], [15,7] ]
链接：
https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof/
解题
在这道题之前有一道题：
https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof/
在上一道题中是把每一层的数据进行打印，在本题是将每层的数据保存到一个list中。
所以只需要在之前的解题上扩展一下思维就好了。
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public List<List<Integer>> levelOrder(TreeNode root) { List<List<Integer>> lists = new ArrayList<>(); //如果等于null，就直接返回一个空的int[]数组 if(root==null) return lists; //利用队列先进先出的特性，保存每一个根节点 Queue<TreeNode> queue = new LinkedList<>(); //先把根入队 queue.offer(root); //队列不为空的时候，一直执行 while (!queue.isEmpty()){ //存储每一层的数据 List<Integer> list = new ArrayList<>(); //把每一层的数据给拿到 queue.size()的大小就是每一层节点的多少 for(int i=queue.size();i>0;i--){ //第一步：取出根元素,存储到list中 list.add(queue.peek().val); //第二步：弹出队列中的根元素 TreeNode pollRoot = queue.poll(); //第三步：分别把根的左子树和右子树入队 if(pollRoot.left!=null) queue.offer(pollRoot.left); if(pollRoot.right!=null) queue.offer(pollRoot.right); } //保存每次的结果 lists.add(list); } // 将lists返回 return lists; } }
