题目地址: leetcode.com/problems/bi…
经典的树的广度优先遍历。用一个队列即可。基本思想是,每当从队列中poll一个数出来的时候,就将它的子节点offer进去,这样每一层遍历完后,队列里已经从左向右存储了当前这一层的所有子节点,于是就可以继续poll出来访问。需要注意的一个地方是,要提前存储队列的size,也就是当前这一层的节点数目,因为队列的size在poll的时候是动态变化的。代码如下:
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;
public class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> rows = new ArrayList<>();
if (root == null) {
return rows;
}
Queue<TreeNode> queue = new ArrayDeque<>();
queue.offer(root);
while (!queue.isEmpty()) {
List<Integer> row = new ArrayList<>();
// 提前存储当前queue的size,也就是当前这一层的节点个数
int len = queue.size();
for (int i = 0; i < len; i++) {
TreeNode x = queue.poll();
row.add(x.val);
if (x.left != null) {
queue.offer(x.left);
}
if (x.right != null) {
queue.offer(x.right);
}
}
rows.add(row);
}
return rows;
}
}
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}