二叉树层次遍历 Java
时间: 2024-02-23 07:55:18 浏览: 93
二叉树的层次遍历是一种广度优先搜索(BFS)的方法,它按照从上到下、从左到右的顺序逐层遍历二叉树的节点。在Java中,可以使用队列来实现二叉树的层次遍历。
下面是一个示例代码,演示了如何实现二叉树的层次遍历:
```java
import java.util.*;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int val) {
this.val = val;
}
}
public class BinaryTreeLevelOrderTraversal {
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);
while (!queue.isEmpty()) {
int levelSize = queue.size();
List<Integer> currentLevel = new ArrayList<>();
for (int i = 0; i < levelSize; i++) {
TreeNode node = queue.poll();
currentLevel.add(node.val);
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
result.add(currentLevel);
}
return result;
}
}
```
在上面的代码中,我们使用一个队列来辅助进行层次遍历。首先将根节点入队,然后进入循环,每次循环开始时,获取当前队列的大小,表示当前层的节点个数。然后依次从队列中取出这些节点,将它们的值加入到当前层的列表中,并将它们的左右子节点(如果存在)入队。循环结束后,将当前层的列表加入到结果列表中。最终返回结果列表即可。
阅读全文