二叉树层序遍历(一)

题目描述

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

该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

提示:
0 <= 二叉树的结点数 <= 1500

示例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
        //存放结果
        ArrayList<ArrayList<Integer>>  rs = new ArrayList<>();

        if(root==null){
            return rs;
        }
        
        
        Queue<TreeNode> q = new ArrayDeque<>();
            q.add(root);
            
        while(!q.isEmpty()){
        
             
            ArrayList<Integer> row = new ArrayList();
            //这里的n是记录二叉树中每层结点的个数
            int n = q.size();
            
            for(int i=0;i<n;i++){
                 //取出队列中的元素放入子集合(row)中
                 TreeNode cur = q.poll();
                 row.add(cur.val);
                 
                 if(cur.left!=null){
                     q.add(cur.left);
                 }
                 if(cur.right!=null){
                     q.add(cur.right);
                 }
                
            }
            rs.add(row);
        }
           return rs;
    }
}