牛客网NC15二叉树的层次遍历

题目

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

该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]
示例2
输入
复制
{1,2,3,4,#,#,5}
返回值
复制
[[1],[2,3],[4,5]]

分析

使用一个队列存放二叉树节点，每次向队列中添加非空节点，从根节点开始加入，将队列中的节点的值取出放入ArrayList中，如果当前节点的左右子节点都不为空，也将放入队列中，依次进行遍历。放入之前需要将当前节点从队列中移除，队列中始终是本层的节点。

代码

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) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
        if(root==null){
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        //根节点入队
        queue.offer(root);
        while(!queue.isEmpty()){
            //存放每层节点个数
            ArrayList<Integer> floor = new ArrayList<Integer>();
            int len = queue.size();
            for(int i=0;i<len;i++){
            //	当前节点出队
                TreeNode node = queue.poll();
                //添加到本层的集合中
                floor.add(node.val);
                //下层节点入队
                if(node.left!=null)
                    queue.offer(node.left);
                if(node.right!=null)
                    queue.offer(node.right);
            }
            //将本层节点的值添加到结果集
            result.add(floor);
        }
        return result;
    }
}