这道题比较中规中矩,就是一个BFS,不过有一个每一层输出顺序的问题(第一层从左到右打印,第二层从右到左打印)
看代码:
public static List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> list = new ArrayList<>();
if (root == null){//根节点空,返回空的list就好了
return list;
}
Queue<TreeNode> queue = new LinkedList<>();//存放每一层节点元素的队列
queue.offer(root);
while (!queue.isEmpty()){
LinkedList<Integer> res = new LinkedList<>();//存放每一层的打印值
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
if(list.size() % 2 ==0){//层数是偶数——>从右向左打印
res.addLast(node.data);
}else {//层数是奇数——>从右向左打印
res.addFirst(node.data);
}
//添加下一层节点
if(node.left!=null){
queue.offer(node.left);
}
if(node.right!=null){
queue.offer(node.right);
}
}
list.add(res);
}
return list;
}
可以看到存储每一层的打印值的时候是用到了一个LinkedList链表,我们知道LinkedList是一个双向链表,对于头尾节点操作很方便,所以如果层数是偶数,需要从右向左打印我们就将左边的节点保存在链表末尾,这样就实现了从右向左打印