题目:https://leetcode-cn.com/problems/search-in-a-binary-search-tree/
思路一:递归
分析:二叉搜索树是一棵二叉树,每个节点都有以下特性:
(1)大于左子树上任意一个节点的值
(2)小于右子树上任意一个节点的。
递归三部曲:
-
确定函数参数和返回值
参数为树的根节点,返回以目标值为根节点的子树
-
确定终止条件
节点为空或者找到目标值的节点就返回
-
单层逻辑
如果访问到的节点的值比目标值大,就访问左子树;如果比目标值小,就访问右子树。节点不存在就返回NULL;
TreeNode* searchBST(TreeNode* root, int val) {
if(root==NULL)
return NULL;
if(root->val==val)
return root;
if(root->val>val)
{
return searchBST(root->left,val);
}
if(root->val<val)
{
return searchBST(root->right,val);
}
return NULL;
}
简洁的写法
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
if(!root) return nullptr;
if(root->val == val) return root;
return (root->val > val) ? searchBST(root->left, val) : searchBST(root->right, val);
}
};
思路二:迭代法
根据二叉搜索树的特性:如果节点值比目标值大,则去左子树找,如果比目标值小,则去右子树找;如果相等就返回;
遍历结束还没找到就返回
TreeNode* searchBST(TreeNode* root, int val) {
while(root!=NULL)
{
if(root->val>val)
root=root->left;
else if(root->val<val)
root=root->right;
else
return root;
}
return NULL;
}