题解 | #牛群分层排列#
牛群分层排列
https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4
知识点:二叉树,层序遍历
要想得到二叉树每一层的值,需要对二叉树进行层序遍历,也就是广度优先搜索,我们需要使用一个队列来实现。具体来讲,我们对于每一层的节点,都需要加入到队列中,当节点出队时,我们将其左右子节点进行入队操作,在遍历每一层节点时,我们使用一个临时的变量size来约束队列出队的个数,也就是当前层的节点个数,确保不会遍历到下一层。
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]; } List<String> list = new ArrayList<>(); Deque<TreeNode> queue = new ArrayDeque<>(); queue.offer(root); while(!queue.isEmpty()) { int size = queue.size(); StringBuilder sb = new StringBuilder(); for(int i = 0; i < size; i++) { TreeNode poll = queue.poll(); sb.append(poll.val); if(poll.left != null) { queue.offer(poll.left); } if(poll.right != null) { queue.offer(poll.right); } } list.add(sb.toString()); } String[] res = new String[list.size()]; for(int i = 0; i < list.size(); i++) { res[i] = list.get(i); } return res; } }