LeetCode—剑指Offer：从上到下打印二叉树Ⅱ（遍历）

从上到下打印二叉树Ⅱ（简单）

2020年8月30日

题目来源：力扣

解题
与从上到下打印二叉树Ⅰ解题思路大体一致，使用队列做bfs，但这道题把结果分层了，需要每次计算当前层次的结果，可以通过计算当前队列的容量。

/**
 * 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) {
        //定义队列
        Queue<TreeNode> queue =new LinkedList<>();
        //定义结果
        List<List<Integer>> res=new ArrayList<>();
        //把根节点进入队列，处理异常情况
        if(root!=null) queue.add(root);
        //当队列不为空时循环
        while(!queue.isEmpty()){
            //定义临时结果
            List<Integer> tmp_res=new ArrayList<>();
            //队列有多少元素就循环多少次，这是分层的关键
            for(int i=queue.size();i>0;i--){
                //出一个节点
                TreeNode node=queue.poll();
                //临时结果存放
                tmp_res.add(node.val);
                if(node.left!=null) queue.add(node.left);
                if(node.right!=null) queue.add(node.right);
            }
            //存储每一层的结果
            res.add(tmp_res);
        }
        return res;
    }
}