题解 | #牛群分层排列#

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) {
        if (root == null) return new String[] {};


        List<String> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);

        while (!queue.isEmpty()) {
            int levelSize = queue.size();  // 记录当前层的节点数
            StringBuilder sb = new StringBuilder();  // 存储当前层的节点值

            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll();
                sb.append(Integer.toString(
                              node.val));  // 将节点值转换为字符串并添加到StringBuilder中

                if (node.left != null) queue.offer(node.left);  // 将左子节点入队
                if (node.right != null) queue.offer(node.right);  // 将右子节点入队
            }

            res.add(sb.toString());  // 将当前层的StringBuilder转换为字符串并添加到结果列表中
        }

        String[] result = new String[res.size()];
        res.toArray(result);
        return result;

    }
}

语言：java

知识点：二叉树的遍历

思路：

通过两个指针来实现，一个指针记录当前层的节点数，另一个指针记录下一层的节点数。 具体步骤如下： 初始化一个队列，将根节点入队。