牛客网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;

    }

}

巴特西

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

题目

分析

代码

最新文章

热门文章