温馨提示×

java treenode如何实现节点搜索

小樊
81
2024-11-28 09:46:40
栏目: 编程语言

在Java中,可以使用递归或迭代方法来实现二叉树的节点搜索。这里我将为您提供一个使用递归方法的示例:

首先,定义一个TreeNode类,用于表示二叉树的节点:

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

接下来,实现一个递归方法searchNode,该方法接受一个TreeNode对象(根节点)和一个目标值作为参数,返回找到的目标节点或null(如果未找到):

public TreeNode searchNode(TreeNode root, int target) {
    if (root == null || root.val == target) {
        return root;
    }

    TreeNode left = searchNode(root.left, target);
    if (left != null) {
        return left;
    }

    return searchNode(root.right, target);
}

这个方法首先检查当前节点是否为null或者节点的值是否等于目标值。如果是,则返回当前节点。否则,它递归地在左子树和右子树中搜索目标值。

以下是一个简单的测试用例:

public static void main(String[] args) {
    TreeNode root = new TreeNode(8);
    root.left = new TreeNode(3);
    root.right = new TreeNode(10);
    root.left.left = new TreeNode(1);
    root.left.right = new TreeNode(6);
    root.right.right = new TreeNode(14);
    root.right.right.left = new TreeNode(13);

    TreeNode result = searchNode(root, 6);
    if (result != null) {
        System.out.println("找到节点,值为: " + result.val);
    } else {
        System.out.println("未找到节点");
    }
}

这个测试用例创建了一个二叉树,并调用searchNode方法搜索值为6的节点。输出结果应为:

找到节点,值为: 6

0