题目描述
思路 1
使用一个容器来存储每一层的节点。然后撤出所有节点后再加入它们的所有子节点。
本来觉得应该建立两个容器,一个容器装当前层,另一个容器一次装入它们的孩子。但是这种做法处理起来很麻烦,浪费空间。
问题的关键是:如何将当前层与子层分开,因此我只需要在当前层和子层中间插入任意一个特殊节点标记为当前层结束即可。
这里,我将这个特殊的标记节点设为 null
代码如下:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
if (root == null) return result;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
queue.offer(null);
List<Integer> layer = new ArrayList<>();
while (!queue.isEmpty()) {
if (queue.peek() == null) {
result.add(layer);
queue.poll();
if (queue.isEmpty()) break;
queue.offer(null);
layer = new ArrayList<>(); // 注意,不能用 layer.clear,否则存储引用的 result 中的数据也会被清空。
continue;
}
TreeNode poll = queue.poll();
layer.add(poll.val);
if (poll.left != null) queue.offer(poll.left);
if (poll.right != null) queue.offer(poll.right);
}
return result;
}
}
思路 2
翻了一下答案,答案中进行分层的思路是:记录每一层的节点的数量。
结合代码解释:
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new ArrayDeque<>();
if (root != null) {
queue.add(root);
}
while (!queue.isEmpty()) { // 每一层循环实际为一层树
int n = queue.size();
List<Integer> level = new ArrayList<>();
for (int i = 0; i < n; i++) { // 将某一层的所有节点全部处理完
TreeNode node = queue.poll();
level.add(node.val);
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
res.add(level);
}
return res;
}
附:
这篇文章还提到了如何使用 BFS 解决无权图最短路径问题以及 BFS 的常规写法。
void bfs(TreeNode root) {
Queue<TreeNode> queue = new ArrayDeque<>();
queue.add(root);
while (!queue.isEmpty()) {
TreeNode node = queue.poll(); // Java 的 pop 写作 poll()
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
}
作者:nettee
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/solution/bfs-de-shi-yong-chang-jing-zong-jie-ceng-xu-bian-l/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
图解十分详细,值得参考。
只是比较两段代码的话,最直观的感受就是:DFS 遍历的代码比 BFS 简洁太多了!这是因为递归的方式隐含地使用了系统的 栈,我们不需要自己维护一个数据结构。如果只是简单地将二叉树遍历一遍,那么 DFS 显然是更方便的选择。
另外文章提到,DFS 通过递归模拟了栈的使用,BFS 需要借助队列。这个规律值得记住。