java实现二叉树层序遍历(附完整源码)
时间: 2023-11-09 09:05:38 浏览: 118
好的,下面是Java实现二叉树层序遍历的完整源码:
```java
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
public class BinaryTreeLevelOrderTraversal {
// 定义二叉树节点类
public static class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
public static 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> levelList = new ArrayList<>();
for (int i = 0; i < levelSize; i++) {
TreeNode node = queue.poll();
levelList.add(node.val);
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
result.add(levelList);
}
return result;
}
public static void main(String[] args) {
// 构建二叉树
TreeNode root = new TreeNode(3);
root.left = new TreeNode(9);
root.right = new TreeNode(20);
root.right.left = new TreeNode(15);
root.right.right = new TreeNode(7);
// 输出层序遍历结果
List<List<Integer>> result = levelOrder(root);
for (List<Integer> levelList : result) {
System.out.println(levelList);
}
}
}
```
这份代码中,我们定义了一个 `TreeNode` 类作为二叉树节点,使用了 `Queue` 类来实现广度优先搜索,最终返回了一个二维列表,其中每个子列表代表一层节点的值。在 `main` 函数中,我们构建了一个二叉树,并输出了其层序遍历的结果。
阅读全文