700 二叉搜索树中的搜索

700 二叉搜索树中的搜索

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 null 。

输入：root = [4,2,7,1,3], val = 2
输出：[2,1,3]
1
2

输入：root = [4,2,7,1,3], val = 5
输出：[]
1
2

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def searchBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        # 非递归1
        # from collections import deque
        # que = deque([root])
        # while que:
        #     node = que.popleft()
        #     if node.val != val:
        #         if node.left:
        #             que.append(node.left)
        #         if node.right:
        #             que.append(node.right)
        #     else:
        #         return node
        # return 
        # 非递归2
        # while root:
        #     if root.val > val:
        #         root = root.left
        #     elif root.val < val:
        #         root = root.right
        #     else: return root
        # return 
        # 递归
        if not root or root.val == val:return root
        if root.val < val:
            return self.searchBST(root.right,val)
        if root.val > val:
            return self.searchBST(root.left, val)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35