题解 | #牛群分层排列#
牛群分层排列
https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4
知识点:二叉树,层序遍历
要想得到二叉树每一层的值,需要对二叉树进行层序遍历,也就是广度优先搜索,我们需要使用一个队列来实现。具体来讲,我们对于每一层的节点,都需要加入到队列中,当节点出队时,我们将其左右子节点进行入队操作,在遍历每一层节点时,我们使用一个临时的变量size来约束队列出队的个数,也就是当前层的节点个数,确保不会遍历到下一层。
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];
}
List<String> list = new ArrayList<>();
Deque<TreeNode> queue = new ArrayDeque<>();
queue.offer(root);
while(!queue.isEmpty()) {
int size = queue.size();
StringBuilder sb = new StringBuilder();
for(int i = 0; i < size; i++) {
TreeNode poll = queue.poll();
sb.append(poll.val);
if(poll.left != null) {
queue.offer(poll.left);
}
if(poll.right != null) {
queue.offer(poll.right);
}
}
list.add(sb.toString());
}
String[] res = new String[list.size()];
for(int i = 0; i < list.size(); i++) {
res[i] = list.get(i);
}
return res;
}
}
深信服公司福利 804人发布
