实现树查询可以使用深度优先搜索 (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 算法查找节点。首先将根节点入队列,然后不断地出队列并将其左右子节点入队列,直到队列为空或者找到目标节点为止。

Java 树查询:深度优先搜索 (DFS) 和广度优先搜索 (BFS) 实现

原文地址: https://www.cveoy.top/t/topic/mskD 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录