二叉搜索树中的搜索

给定二叉搜索树(BST)的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 NULL。

例如,

给定二叉搜索树:

    4
   / \
  2   7
 / \
1   3

和值: 2
你应该返回如下子树:

  2     
 / \   
1   3

在上述示例中,如果要找的值是 5,但因为没有节点值为 5,我们应该返回 NULL。

通过次数119,560提交次数154,482

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-a-binary-search-tree
解题方法:

   @Data
    static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    static class Solution {
        public static TreeNode searchBST(TreeNode root, int val) {
        //循环
//            while (root != null && root.val != val) {
//                root = root.val > val ? root.left : root.right;
//            }
//            System.out.println(root.val);
//            return root;
//递归
            if (root == null) {
                return null;
            }
            if (root.val == val) {
                return root;
            }
            return val < root.val ? searchBST(root.left, val) : searchBST(root.right, val);

        }
    }

    public static TreeNode treeSet() {
        TreeNode root = new TreeNode(4);
        TreeNode a = new TreeNode(2);
        TreeNode b = new TreeNode(7);
        TreeNode c = new TreeNode(1);
        TreeNode d = new TreeNode(3);
        root.left = a;
        root.right = b;
        a.left = c;
        a.right = d;
        return root;
    }
        public static void main(String[] args) {
        TreeNode treeNode = UserUtil.treeSet();
        TreeNode treeNode1 = Solution.searchBST(treeNode, 2);
        int val = treeNode1.getLeft().getVal();
        int val1 = treeNode1.getRight().getVal();
        System.out.println(treeNode1.getVal());
        System.out.println(val);
        System.out.println(val1);
            }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值