leetcode700.二叉搜索树中的搜索
给定二叉搜索树(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 输出:[]
代码
//leetcode700 二叉搜索树中的搜索
//迭代法、递归同理
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
TreeNode* cur = root;
while (cur != nullptr)
{
if (cur->val == val)
{
return cur;
}
else if (cur->val > val)
{
cur = cur->left;
}
else
{
cur = cur->right;
}
}
return nullptr;
}
};
leetcode98.验证二叉搜索树
给你一个二叉树的根节点 root
,判断其是否是一个有效的二叉搜索树。
有效 二叉搜索树定义如下:
- 节点的左子树只包含 小于 当前节点的数。
- 节点的右子树只包含 大于 当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
示例 1:
输入:root = [2,1,3] 输出:true示例 2:
输入:root = [5,1,4,null,null,3,6] 输出:false 解释:根节点的值是 5 ,但是右子节点的值是 4 。
代码
//leetcode98 验证二叉搜索树
// 中序遍历看看是不是递增序列
class Solution {
public:
bool isValidBST(TreeNode* root) {
long long preNum = LLONG_MIN;
stack<TreeNode*> treeSta;
TreeNode* cur = root;
while (!treeSta.empty() || cur != nullptr)
{
while (cur != nullptr)
{
treeSta.push(cur);
cur = cur->left;
}
cur = treeSta.top();
treeSta.pop();
if (preNum >= cur->val)
{
return false;
}
preNum = cur->val;
cur = cur->right;
}
return true;
}
};