来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/validate-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
98.给定一个二叉树,判断其是否是一个有效的二叉搜索树。
假设一个二叉搜索树具有如下特征:
节点的左子树只包含小于当前节点的数。
节点的右子树只包含大于当前节点的数。
所有左子树和右子树自身必须也是二叉搜索树。
- 看到题目时,想着跟平常解决二叉树题目一样,递归地判断左右子树是否满足二叉搜索树的要求,于是有了以下代码。可惜到提交的时候却出了错,才发现以下代码忽略了一种情况,那就是局部的子树满足二叉搜索树的情况,但是从整体上看并不满足,比如左子树出现比根节点还要大的元素,右子树出现比根节点还要小的元素。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
bool isValidBST(struct TreeNode* root){
if (root == NULL || (root->left == NULL && root->right == NULL))
return true;
if (root->left != NULL && root->left->val >= root->val)
return false;
if (root->right != NULL && root->right->val <= root->val)
return false;
bool isLeft = isValidBST(root->left);
bool isRight = isValidBST(root->right);
return isLeft && isRight;
}
- 上面这种解法解决不了问题,就需要寻求新的解决方案了。由于二叉搜索树的中序遍历序列是有序的,因此可以通过递归得到给定二叉树的中序遍历序列,再判断是否有序即可。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
void traversal(struct TreeNode* root, int* inorder, int* i) {
if (root == NULL)
return;
traversal(root->left, inorder, i);
inorder[(*i)++] = root->val;
traversal(root->right, inorder, i);
}
bool isValidBST(struct TreeNode* root){
int* inorder = malloc(sizeof(int) * 10000);
int i = 0;
traversal(root, inorder, &i);
for (int j = 0; j < i - 1; j++) {
if (inorder[j] >= inorder[j + 1])
return false;
}
return true;
}
- 也可以在中序遍历的过程中验证遍历序列是否按照升序的顺序排列,则需要用到一个全局变量来存储当前遍历结点的前一个遍历结点。奇怪的是照着题解写出来的代码一直出错,卡在了测试用例[0]这里,不知道是什么情况。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
struct TreeNode* pre = NULL;
bool isValidBST(struct TreeNode* root){
if (root == NULL)
return true;
bool isLeft = isValidBST(root->left);
if (pre != NULL && (pre->val >=root->val))
return false;
pre = root;
bool isRight = isValidBST(root->right);
return isLeft && isRight;
}
- 还有迭代法解法,只需要对中序遍历的迭代法进行改进即可。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
bool isValidBST(struct TreeNode* root){
struct TreeNode* stack[2000];
int top = -1;
struct TreeNode* cur = root;
struct TreeNode* pre = NULL;
while (cur != NULL || top != -1) {
if (cur != NULL) {
stack[++top] = cur;
cur = cur->left;
} else {
cur = stack[top--];
if (pre != NULL && pre->val >= cur->val)
return false;
pre = cur;
cur = cur->right;
}
}
return true;
}