题解 | #牛群分层排列#
牛群分层排列
https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4
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
知识点:二叉树的遍历
思路:
通过两个指针来实现,一个指针记录当前层的节点数,另一个指针记录下一层的节点数。 具体步骤如下: 初始化一个队列,将根节点入队。