题解 | #牛群分层排列#

题目考察的知识点

考察二叉树的广度优先遍历

题目解答方法的文字分析

具体做法要借助于队列，队列中存储的是下个将要访问的节点。当前节点入队之后，节点值加入字符串中，随后从队中丢弃该节点，去遍历该节点的子节点并入队。具体见代码注释。

本题解析所用的编程语言

使用Java解答

完整且正确的编程代码

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return string字符串一维数组
     */
    public String[] levelOrder (TreeNode root) {
        // write code here
        List<String> ret = new ArrayList<String>();
        if (root == null) {
            return new String[0];
        }

        Queue<TreeNode> queue = new LinkedList<TreeNode>(); //队列 数据类型是TreeNode
        queue.offer(root); //加入当前节点
        while (!queue.isEmpty()) { //遍历当前节点的所有子节点
            String str = "";
            int currentLevelSize = queue.size();
            for (int i = 1; i <= currentLevelSize; ++i) {
                TreeNode node = queue.poll(); //将当前节点从队列中丢弃
                str+=node.val;
                if (node.left != null) {
                    queue.offer(node.left); //子节点入队
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            ret.add(str);
        }
        String[] res = new String[ret.size()];
        for(int i=0; i<ret.size(); i++){
            res[i] = ret.get(i);
        }
        return res; 
    }   
}