700. 二叉搜索树中的搜索

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

例如，

给定二叉搜索树:

4
/
2 7
/
1 3

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

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

通过次数92,931提交次数121,629

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

题解

Swift

public class TreeNode {
public var val: Int
public var left: TreeNode?
public var right: TreeNode?
public init() { self.val = 0; self.left = nil; self.right = nil }
public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil }
public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
self.val = val
self.left = left
self.right = right
}
}

// 2利用二叉搜索特性
class Solution {
func searchBST(_ root: TreeNode?, _ val: Int) -> TreeNode? {
if let rVal = root?.val {
if rVal == val {
return root
} else if rVal > val {
return searchBST(root?.left, val)
} else if rVal < val {
return searchBST(root?.right, val)
}
}

return nil
}
}

//// 1暴力解法
// class Solution {
// func searchBST(_ root: TreeNode?, _ val: Int) -> TreeNode? {
// guard root != nil else {
// return nil
// }
//
// if root?.val == val {
// return root
// }
//
// let left = searchBST(root?.left, val)
//
// if left != nil {
// return left
// }
// let right = searchBST(root?.right, val)
//
// if right != nil {
// return right
// }
//
// return nil
// }
// }

let node1 = TreeNode(1)
let node3 = TreeNode(3)
let node2 = TreeNode(2, node1, node3)
let node7 = TreeNode(7)
let node4 = TreeNode(4, node2, node7)

print(Solution().searchBST(node4, 2)?.val as Any)