用Java实现树的深度优先算法和广度优先算法
时间: 2023-07-26 07:39:44 浏览: 117
树的深度优先算法和广度优先算法都是常见的树遍历算法,下面是Java实现的示例代码:
```java
import java.util.*;
// 树节点类
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int val) {
this.val = val;
}
}
// 树遍历类
class TreeTraversal {
// 树的深度优先遍历算法
public List<Integer> dfs(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pop();
res.add(node.val);
if (node.right != null) {
stack.push(node.right);
}
if (node.left != null) {
stack.push(node.left);
}
}
return res;
}
// 树的广度优先遍历算法
public List<Integer> bfs(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int levelSize = queue.size();
for (int i = 0; i < levelSize; i++) {
TreeNode node = queue.poll();
res.add(node.val);
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
}
return res;
}
}
// 测试类
public class TreeTraversalTest {
public static void main(String[] args) {
// 构造一棵二叉树
TreeNode root = new TreeNode(1);
root.left = new TreeNode(2);
root.right = new TreeNode(3);
root.left.left = new TreeNode(4);
root.left.right = new TreeNode(5);
root.right.left = new TreeNode(6);
root.right.right = new TreeNode(7);
// 深度优先遍历
TreeTraversal treeTraversal = new TreeTraversal();
System.out.println("深度优先遍历:" + treeTraversal.dfs(root));
// 广度优先遍历
System.out.println("广度优先遍历:" + treeTraversal.bfs(root));
}
}
```
在上面的代码中,我们首先定义了一个`TreeNode`类,表示树节点。然后,我们定义了一个`TreeTraversal`类,其中包含树的深度优先遍历算法和广度优先遍历算法。在深度优先遍历算法中,我们使用了一个栈来实现遍历过程;在广度优先遍历算法中,我们使用了一个队列来实现遍历过程。最后,我们在`TreeTraversalTest`类中构造了一棵二叉树,并分别调用了深度优先遍历和广度优先遍历算法进行测试。
阅读全文
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20241231044930.png)
![-](https://img-home.csdnimg.cn/images/20241231044930.png)
![rar](https://img-home.csdnimg.cn/images/20241231044955.png)
![-](https://img-home.csdnimg.cn/images/20241231045053.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![java](https://img-home.csdnimg.cn/images/20250102104920.png)
![zip](https://img-home.csdnimg.cn/images/20241231045053.png)
![rar](https://img-home.csdnimg.cn/images/20241231044955.png)
![-](https://img-home.csdnimg.cn/images/20241231044955.png)
![-](https://img-home.csdnimg.cn/images/20241231044955.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![txt](https://img-home.csdnimg.cn/images/20241231045021.png)
![-](https://img-home.csdnimg.cn/images/20241231044955.png)
![txt](https://img-home.csdnimg.cn/images/20241231045021.png)