以 LeetCode102 作为例子:
题目描述
思路描述
层序遍历需要用到的数据结构是队列。需要考虑的问题是:如何标识当前节点的层数。
有以下三种方法:
方法 1
将每个节点表示为一个二元组 (node, level),这种方法效率太低,不考虑。感兴趣可以参考
方法 2
遍历完一层节点后,在队列中插入一个标记节点NULL
,这个标记节点没有具体意义,只是标识某一层已经遍历结束。
这种方法的缺点在于,假如想要在层序遍历过程中,有元素为 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<>();
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;
}
}
方法 3
方法 3 是按层进行操作队列,每次循环不是取出一个节点,而是取出一整层节点。
我们可以用一种巧妙的方法修改 BFS:
- 首先根元素入队
- 当队列不为空的时候
- 求当前队列的长度 s_is
- 依次从队列中取 s_is 个元素进行拓展,然后进入下一次迭代
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/solution/er-cha-shu-de-ceng-xu-bian-li-by-leetcode-solution/
来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
代码如下:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ret = new ArrayList<List<Integer>>();
if (root == null) {
return ret;
}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
while (!queue.isEmpty()) {
List<Integer> level = new ArrayList<Integer>();
int currentLevelSize = queue.size();
for (int i = 1; i <= currentLevelSize; ++i) {
TreeNode node = queue.poll();
level.add(node.val);
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
ret.add(level);
}
return ret;
}
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/solution/er-cha-shu-de-ceng-xu-bian-li-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。