题解 | #牛群分层排列#
牛群分层排列
https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4
本题知识点分析:
- 二叉树的层序遍历
- 字符串的拼接
问题分析:
- 如果给定的根节点 root 为空,直接返回一个空的字符串数组。
- 先创建一个 ArrayList 类型的集合 list,用于存储遍历结果。
- 创建一个 LinkedList 类型的队列 queue,用于存储待遍历的节点。
- 如果根节点不为空,加入到队列中。
- 当队列不为空时,遍历队列中节点,并将节点值拼接到sb中,如左右子树不为空,添加到queue中
- 将拼接好的字符串放入list集合
- 最后将集合list转换成字符串数组,并返回结果。
本题使用编程语言: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<>();
List<String> list = new ArrayList<>();
StringBuffer sb = new StringBuffer();
TreeNode temp;
int count = 1;
int cycle;
queue.add(root);
while (!queue.isEmpty()) {
cycle = count;
count = 0;
for (int i = 0; i < cycle; i++) {
temp = queue.poll();
sb.append(temp.val);
if (temp.left != null) {
queue.add(temp.left);
count++;
}
if (temp.right != null) {
queue.add(temp.right);
count++;
}
}
list.add(sb.toString());
sb.delete(0,sb.length());
}
String[] res = new String[list.size()];
for(int i = 0;i< list.size();++i){
res[i] = list.get(i);
}
return res;
}
}
OPPO公司福利 1202人发布