LeetCode
- 题目地址:https://leetcode.com/problems/validate-binary-search-tree/?tab=Description
- 思考过程:
- 题目要求是给一个树的根节点,判断是否二叉搜索树,即左子树最大的小于当前根节点的值,当前根节点的值小于右子树最小的。一开始想的花里胡哨,想弄两个递归函数一个求最大一个求最小,但是这样的话对每个节点都要求一次显得很困难,已经遍历过的信息没办法给它的子树去用。
- 所以换了一种思路,二叉搜索树其实就是按照中序遍历的时候,数值从大到小,那么只需要:
- 对于最左边的数,将它设置为pre
- 中序遍历剩下的数,与pre比较,如果pre大于等于(题目要求严格后面大于前面的)当前值则返回false
- 左子树先遍历,如果不满足BST返回false;
- 遍历当前节点有三种情况:
- 还没设置pre,那么设置pre并且把isSet置为true
- 设置了pre,那么和pre作比较,如果pre大于等于当前节点的值,返回false
- pre小于当前节点的值,则pre = root->val
- 右子树再遍历,最后函数返回右子树遍历的结果(右子树为true,则以当前节点为根的树都满足BST返回true,右子树为false,则说明右子树不满足BST,所以当前根的树不满足BST,返回false)
- 设置isSet = false,对根调用递归即可判断是否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 Solution {
public:
bool isValidBST(TreeNode* root) {
isSet = false;
return rec(root);
}
bool rec(TreeNode* root) {
if (root == NULL)
return true;
bool left = rec(root->left);
if (left == false)
return false;
if (!isSet) {
pre = root->val;
isSet = true;
}
else if (pre >= root->val)
return false;
else
pre = root->val;
bool right = rec(root->right);
return right;
}
private:
int pre;
bool isSet;
};