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];
}
// 创建一个动态列表 res,用于存储层序遍历结果
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 中
res.add(sb.toString());
}
// 将动态列表 res 转为字符串数组 ans
String[] ans = new String[res.size()];
int i = 0;
for (String s : res) {
ans[i++] = s;
}
// 返回结果数组
return ans;
}
}
该题考察的知识点是二叉树的层序遍历。
层序遍历是一种广度优先搜索(BFS)的方式,通过逐层访问二叉树的节点,按照从左到右的顺序输出节点值。
算法思路如下:
- 如果根节点为空,直接返回空数组。
- 创建一个动态列表
res,用于存储层序遍历结果。 - 创建一个队列
queue,用于层序遍历。将根节点加入队列。 - 循环遍历队列,直到队列为空。创建一个字符串构建器 sb,用于拼接当前层级的节点值。获取当前层级的节点个数,使用一个变量 count 记录。遍历当前层级的节点,每次取出队列头部节点。将节点值加入到字符串构建器 sb 中。如果当前节点的左子节点不为空,将左子节点加入队列。如果当前节点的右子节点不为空,将右子节点加入队列。将字符串构建器 sb 转为字符串,并将结果添加到列表 res 中。
- 将动态列表
res转为字符串数组ans,并返回结果数组。

京公网安备 11010502036488号