每日一题力扣700 二叉搜索树中的搜索

 

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

例如，

给定二叉搜索树:

4
/ \
2 7
/ \
1 3

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

2
/ \
1 3

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-in-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二叉搜索树的特点就是左＜=根＜=右，所以在找的时候，像二分一样搜索

class Solution:
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
        def bin_search(root, val):
            if not root:
                return 
            if root.val == val:
                return root
            elif root.val > val:
                return bin_search(root.left, val)
            else:
                return bin_search(root.right, val)
        return bin_search(root, val)