题解 | #牛群分层排列#
牛群分层排列
https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4
- 题目考察的知识点
二叉树的层次遍历
- 题目解答方法的文字分析
如果root==null,则返回空的字符串数组。否则,将二叉树每一层的字符串记录下来。这个主要使用二叉树的层次遍历实现。先遍历第一层根结点,将其左右非空孩子先后加入队列,利用队列先进先出的特点,使得同一层的节点依次输出,从而保证按层遍历二叉树。为了保证能够获得每一层的二叉树节点值组成的字符串,我们每次在进行下一层次的节点入队之前,先将上一层的节点个数记录下来,并记为count,由于队列是先进先出,所以,前count个节点就是上一层的所有节点。并用StringBuilder来记录每一层的字符串。最后再将字符串列表转换成字符串数组即可
- 本题解析所用的编程语言
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) {
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;
}
}