题目描述
今天我们来看的是二叉搜索树中的搜索和插入操作,力扣题P700、701。
来看题目。
题目讲解
搜索
首先看搜索题,即p700。
这个是偏简单的题目,只需一个一个寻找就行,若没找到便返回空。
在这里我使用了一个while循环,还使用了 > ? :语句,就是关系运算符成立便执行:之前的语句,否则反之。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution
{
public:
TreeNode* searchBST(TreeNode* root, int val)
{
while(root)
{
if(root->val == val)
{
return root;
}
root = root->val > val ? root->left :root->right;
}
return nullptr;
}
};
插入操作
再来看p701,这次是个中等题,看题目介绍,插入方式有多种,目前我是用的插在叶节点哈。
这次我们先判断它是否为空,为空边直接插入,当前节点的值直接变成插入的值,否则我们寻找这个值应该插入的位置。
class Solution
{
public:
TreeNode* insertIntoBST(TreeNode* root, int val)
{
if (root == nullptr) //判断是否为空节点
{
return new TreeNode(val);
}
TreeNode* find = root;
while (find != nullptr)
{
if (val < find->val)
{
if (find->left == nullptr)
{
find->left = new TreeNode(val);
break;
}
else
{
find = find->left;
}
}
else
{
if (find->right == nullptr)
{
find->right = new TreeNode(val);
break;
}
else
{
find = find->right;
}
}
}
return root;
}
};
题目总结
这两个题目还是算偏简单的题目,有待继续练习。