给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
思路:利用队列先进先出的特点,将二叉树的结点存储进队列中,另外使用 List 存储结点的数据。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> list = new ArrayList<>();
if(root == null){
return list;
}
Queue<TreeNode> queue = new ArrayDeque<>();
queue.offer(root);
while(!queue.isEmpty()){
int levelSum = queue.size();
List<Integer> subList = new ArrayList<>();
for(int i = 0;i < levelSum ;i++){
TreeNode node = queue.poll();
subList.add(node.val);
if(node.left != null){
queue.offer(node.left);
}
if(node.right!=null){
queue.offer(node.right);
}
}
list.add(subList);
}
return list;
}
}
队列:Queue 继承了 Collection 接口,其实现类有 Deque, LinkedList, PriorityQueue, BlockingQueue。
应用场景:缓冲,并发访问。
主要方法:查询,添加,删除都有两个方法
- 添加:add(E) 添加失败会报错(队列已满的情况),offer(E) 添加失败返回 false。
- 删除:remove(E) 当队列为空时,会报错,poll(E) 则会返回 null (获取并删除)。
- 查询(获取不删除):element(E) 为空会抛出异常,peek(E) 则会返回 null。