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)的方式,通过逐层访问二叉树的节点,按照从左到右的顺序输出节点值。

算法思路如下:

  1. 如果根节点为空,直接返回空数组。
  2. 创建一个动态列表 res,用于存储层序遍历结果。
  3. 创建一个队列 queue,用于层序遍历。将根节点加入队列。
  4. 循环遍历队列,直到队列为空。创建一个字符串构建器 sb,用于拼接当前层级的节点值。获取当前层级的节点个数,使用一个变量 count 记录。遍历当前层级的节点,每次取出队列头部节点。将节点值加入到字符串构建器 sb 中。如果当前节点的左子节点不为空,将左子节点加入队列。如果当前节点的右子节点不为空,将右子节点加入队列。将字符串构建器 sb 转为字符串,并将结果添加到列表 res 中。
  5. 将动态列表 res 转为字符串数组 ans,并返回结果数组。