思路

层序遍历就是一层一层的去遍历,如上图就是:1 - 2 - 4 - 5 - 3.
这道题要求的返回值时 ArrayList<ArrayList<integer>>。说明每一层都要用一个 ArrayList 去存储,大家想想,我想要有序的去层序遍历应该通过什么数据结构是其有序的去访问那?答案是:队列。
咱们可以将元素每一层按从左到右的顺序去存储到队列中,然后每一层存储到一个 ArrayList 中即可,接下来看看代码吧。</integer>
AC 代码
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
// write code here
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
if (root == null) {
return result;
}
// 队列,用于存储元素
Queue<TreeNode> queue = new LinkedList<>();
// 根节点先入队
queue.offer(root);
// 当队列不为空的时候
while(!queue.isEmpty()) {
// 队列的大小就是这一层的元素数量
int size = queue.size();
ArrayList<Integer> list = new ArrayList<>();
// 开始遍历这一层的所有元素
for (int i = 0; i < size; i ++) {
TreeNode node = queue.poll();
// 如果左节点不为空,则入队,作为下一层来遍历
if(node.left != null) {
queue.offer(node.left);
}
// 同上
if (node.right != null) {
queue.offer(node.right);
}
// 存储一层的节点
list.add(node.val);
}
// 将一层所有的节点汇入到总的结果集中
result.add(list);
}
return result;
}
时间复杂度:O(n),n 为节点数量
空间复杂度:O(n),n 为节点数量
最后
微信搜索:【蘑菇睡不着】关注我的公众号,一起学习。

京公网安备 11010502036488号