给定二叉搜索树(BST)的根节点
root
和一个整数值val
。你需要在 BST 中找到节点值等于
val
的节点。 返回以该节点为根的子树。 如果节点不存在,则返回null
。示例 1:
输入:root = [4,2,7,1,3], val = 2 输出:[2,1,3]示例 2:
输入:root = [4,2,7,1,3], val = 5 输出:[]提示:
- 数中节点数在
[1, 5000]
范围内1 <= Node.val <= 107
root
是二叉搜索树1 <= val <= 107
class Solution {
public:
TreeNode *tmp=nullptr;
TreeNode* searchBST(TreeNode* root, int val) {
if(root->val>val&&root->left)
searchBST(root->left,val);
if(root->val<val&&root->right)
searchBST(root->right,val);
if(root->val==val)
{
tmp=root;
return tmp;
}
return tmp;
}
};
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
TreeNode *tmp=nullptr;
if(root==nullptr||root->val==val)
return root;
if(root->val>val&&root->left)
tmp=searchBST(root->left,val);
if(root->val<val&&root->right)
tmp=searchBST(root->right,val);
return tmp;
}
};
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
if(root==nullptr||root->val==val)
return root;
if(root->val>val)
return searchBST(root->left,val);
if(root->val<val)
return searchBST(root->right,val);
return nullptr;
}
};
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
while(root)
{
if(root->val>val)
root=root->left;
else if(root->val<val)
root=root->right;
else
return root;
}
return nullptr;
}
};