题解 | #牛群分层排列#
牛群分层排列
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[]{}; //使用队列,用于存放临时节点 Queue<TreeNode> q = new LinkedList<>(); q.offer(root); //定义一个链表,用于存放结果集,由于返回类型是数组,数组又是定长的,我们暂时不知道数组的长度,所以先用链表存放结果 List<String> list = new ArrayList<>(); //队列不为空的之后就一直遍历 while(!q.isEmpty()){ //记录当前队列存入的个数,此为一层的数据 int size = q.size(); //使用动态字符串来存储每一层的数据 StringBuilder level = new StringBuilder(); for(int i=0;i<size;i++){ TreeNode node = q.poll(); if(node.left != null){ q.offer(node.left); } if(node.right != null){ q.offer(node.right); } level.append(node.val); } list.add(level.toString()); } String[] res = new String[list.size()]; //遍历封装结果集, for(int i=0;i<list.size();i++){ res[i] = list.get(i); } return res; } }
层序遍历