Java 树查询:深度优先搜索 (DFS) 和广度优先搜索 (BFS) 实现
实现树查询可以使用深度优先搜索 (DFS) 或广度优先搜索 (BFS) 算法。
下面是一个简单的示例代码,使用 DFS 算法查询树中是否存在某个节点:
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val) {
this.val = val;
}
}
public class TreeSearch {
public boolean search(TreeNode root, int target) {
if (root == null) {
return false;
}
if (root.val == target) {
return true;
}
return search(root.left, target) || search(root.right, target);
}
}
在上面的代码中,search 方法采用递归的方式查找节点。如果当前节点为 null,则返回 false;如果当前节点的值等于目标值,则返回 true;否则继续递归查找左子树和右子树。
如果要使用 BFS 算法查询树中是否存在某个节点,可以使用队列来实现:
public boolean search(TreeNode root, int target) {
if (root == null) {
return false;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
if (node.val == target) {
return true;
}
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
return false;
}
在上面的代码中,search 方法采用 BFS 算法查找节点。首先将根节点入队列,然后不断地出队列并将其左右子节点入队列,直到队列为空或者找到目标节点为止。
原文地址: https://www.cveoy.top/t/topic/mskD 著作权归作者所有。请勿转载和采集!