验证二叉搜索树
给定一个二叉树,判断其是否是一个有效的二叉搜索树。
假设一个二叉搜索树具有如下特征:
- 节点的左子树只包含小于当前节点的数。
- 节点的右子树只包含大于当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
示例 1:
输入:
2
/ \
1 3
输出: true
示例 2:
输入:
5
/ \
1 4
/ \
3 6
输出: false
解释: 输入为: [5,1,4,null,null,3,6]。
根节点的值为 5 ,但是其右子节点值为 4 。
遍历时记录一个当前允许的最大值和最小值。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool dfs(TreeNode *root, int min, int max)
{
if(!root)
return true;
if(root->val < min || root->val > max)
return false;
return dfs(root->left, min, root->val-1) && dfs(root->right, root->val+1, max);
}
bool isValidBST(TreeNode* root) {
return dfs(root, LONG_MIN, LONG_MAX);
}
};
第二种方法:使用中序遍历二叉树,然后判断是否是升序序列
class Solution {
public:
void inorder(TreeNode *root, vector<int> &ret)
{
if(root)
{
inorder(root->left, ret);
ret.push_back(root->val);
inorder(root->right, ret);
}
}
bool isValidBST(TreeNode* root) {
if(root == NULL)
return true;
vector<int>ret;
inorder(root, ret);
for(int i = 0; i < ret.size()-1; i++)
{
if(ret[i] >= ret[i+1])
return false;
}
return true;
}
};
二叉搜索树迭代器
实现一个二叉搜索树迭代器。你将使用二叉搜索树的根节点初始化迭代器。
调用 next()
将返回二叉搜索树中的下一个最小的数。
示例:
BSTIterator iterator = new BSTIterator(root);
iterator.next(); // 返回 3
iterator.next(); // 返回 7
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 9
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 15
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 20
iterator.hasNext(); // 返回 false
提示:
next()
和hasNext()
操作的时间复杂度是 O(1),并使用 O(h) 内存,其中 h 是树的高度。- 你可以假设
next()
调用总是有效的,也就是说,当调用next()
时,BST 中至少存在一个下一个最小的数。
这里空间复杂度为树的高度,所以用一个栈来辅助遍历二叉搜索树
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class BSTIterator {
private:
stack<TreeNode*> inorder;
public:
BSTIterator(TreeNode* root) {
while(root)
{
inorder.push(root);
root = root->left;
}
}
/** @return the next smallest number */
int next() {
TreeNode *node = inorder.top();
inorder.pop();
TreeNode *right = node->right;
while(right)
{
inorder.push(right);
right = right->left;
}
return node->val;
}
/** @return whether we have a next smallest number */
bool hasNext() {
return !inorder.empty();
}
};
/**
* Your BSTIterator object will be instantiated and called as such:
* BSTIterator* obj = new BSTIterator(root);
* int param_1 = obj->next();
* bool param_2 = obj->hasNext();
*/