剑指offer32-2：从上到下分行打印二叉树

1. 题目描述

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

输出：[[1], [2, 3], [4, 5]]

即在上题的基础上，分行打印。

2. 解题思路

在上一题层序打印二叉树的基础上，在while中加入for循环，循环次数为该层的节点数：将该层的节点值先存入容器1中。for循环结束后，再把容器1放进容器2中，即可得到满足题目要求的输出。

3. 代码

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class ques32_2 {

    public static List<List<Integer>> levelOrder2(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> res = new ArrayList<>();
        if(root != null) queue.add(root);
        //直到队列为空为止
        while(!queue.isEmpty()) {
            List<Integer> tmp = new ArrayList<>();
            //循环长度为当前队列的长度，即每一层的个数
            for(int i = queue.size(); i > 0; i--) {
                //返回并移除队列的头元素
                TreeNode node = queue.poll();
                tmp.add(node.val);
                if(node.left != null) queue.add(node.left);
                if(node.right != null) queue.add(node.right);
            }
            res.add(tmp);
        }
        return res;
    }


    public static void main(String[] args) {
        //构建树
        TreeNode a4 = new TreeNode(4);
        TreeNode a5 = new TreeNode(5);
        TreeNode a3 = new TreeNode(3);
        a3.left = a4;
        a3.right = a5;
        TreeNode a2 = new TreeNode(2);
        TreeNode a1 = new TreeNode(1);
        a1.left = a2;
        a1.right = a3;

        //测试
        List<List<Integer>> res  = levelOrder2(a1);
        for (int i = 0; i < res.size(); i++) {
            for (int j = 0; j < res.get(i).size(); j++) {
                System.out.println(res.get(i).get(j));
            }
            System.out.println();
        }
    }

    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) { val = x; }
    }
}