题解 | #牛群分层排列#

牛群分层排列

https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4

  1. 题目考察的知识点

二叉树的层次遍历

  1. 题目解答方法的文字分析

如果root==null,则返回空的字符串数组。否则,将二叉树每一层的字符串记录下来。这个主要使用二叉树的层次遍历实现。先遍历第一层根结点,将其左右非空孩子先后加入队列,利用队列先进先出的特点,使得同一层的节点依次输出,从而保证按层遍历二叉树。为了保证能够获得每一层的二叉树节点值组成的字符串,我们每次在进行下一层次的节点入队之前,先将上一层的节点个数记录下来,并记为count,由于队列是先进先出,所以,前count个节点就是上一层的所有节点。并用StringBuilder来记录每一层的字符串。最后再将字符串列表转换成字符串数组即可

  1. 本题解析所用的编程语言

java

  1. 完整且正确的编程代码
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) {
        if(root ==null){
            return new String[0];
        }
        List<String> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()){
            StringBuilder sb =new StringBuilder();
            int count = queue.size();
            while(count-- >0){
                TreeNode node = queue.poll();
                sb.append(node.val+"");
                if (node.left != null) {
					queue.add(node.left);
				}
				if (node.right != null) {
					queue.add(node.right);
				}
            }
            res.add(sb.toString());
        }

        String[] ans =new String[res.size()];
        int i=0;
        for(String s:res){
            ans[i++] = s;
        }
        return ans;
    }
}
全部评论

相关推荐

1 收藏 评论
分享
牛客网
牛客企业服务