题解 | #牛群分层排列#
牛群分层排列
https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4
知识点:二叉树 层序遍历
思路:送分题,层序遍历,一个点,如果判断是这一层的叶子?判断队列的大小即可,
比如 1 2 3 ,送入2,3,判断队列大小,2,就是第二层的叶子
编程语言:java
如果我的思路启发了你,给个小小关注吧~
我是废江,一个从java跑到内核再准备润回java的打工人,我会持续分享从linux内核到上层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
if(root == null)
return new String[0];
Queue<TreeNode> queue = new LinkedList<>();
ArrayList<String> array = new ArrayList<>();
queue.add(root);
while (queue.size() != 0) {
StringBuilder str = new StringBuilder();
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
str.append(node.val);
if (node.left != null)
queue.add(node.left);
if (node.right != null)
queue.add(node.right);
}
array.add(str.toString());
}
return (String[]) array.toArray(new String[0]);
}
}