NC15 求二叉树的层序遍历

题目链接：https://www.nowcoder.com/practice/04a5560e43e24e9db4595865dc9c63a3

题目描述

给定一个二叉树，返回该二叉树层序遍历的结果，（从左到右，一层一层地遍历）
例如：
给定的二叉树是{3,9,20,#,#,15,7},

该二叉树层序遍历的结果是

[

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

]

数据范围：二叉树的节点数满足 1≤n≤10^5

示例 1：

输入：{1,2}
返回值：[[1],[2]]

示例 2：

输入：{1,2,3,4,#,#,5}
返回值：[[1],[2,3],[4,5]]

解题代码

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
        // write code here
        if (root == null) return new ArrayList<>();
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        //广度优先算法
        //使用队列,每次记录一个层级,根据queue的size,来判定当前层级的元素个数.
        Queue<TreeNode> queue = new LinkedList<>();
        ArrayList<Integer> tList;
        queue.add(root);
        while (!queue.isEmpty()) {
            tList = new ArrayList<>();
            int nodeNum = queue.size();
            for (int i = 0; i < nodeNum; i++) {
                TreeNode temp = queue.poll();
                //队列头部元素的左右节点
                if (temp.left != null) {
                    queue.add(temp.left);
                }
                if (temp.right != null) {
                    queue.add(temp.right);
                }
                tList.add(temp.val);
            }
            list.add(tList);
        }
        return list;
    }
}