leetcode:Validate Binary Search Tree

来源:互联网 发布:手机淘宝3.9 编辑:程序博客网 时间:2024/05/27 06:12

最开始我犯了个错误,直接递归判断根节点的值是否在左孩子节点和右孩子节点值得中间,忽略了应该是根节点的值大于左子树所有节点的值,小于右子树所有节点的值,导致代码如下:

bool isValidBST(TreeNode* root) {        if(root==NULL)            return true;        else if(root->left&&!root->right)            return isValidBST(root->left)&&(root->val>root->left->val);        else if(!root->left&&root->right)            return isValidBST(root->right)&&(root->val<root->right->val);        else if(root->left&&root->right)            return isValidBST(root->left)&&isValidBST(root->right)&&(root->val>root->left->val)&&(root->val<root->right->val);        else            return true;    }

这样当测试用例为{10,5,15,#,#,6,20}就会出错。所以在递归函数中应该加入更多的信息,可以记录当前根节点的取值区间,在递归计算时更新取值区间,这样代码如下:

bool check(TreeNode* root,long leftval, long rightval)    {        if(!root)            return true;        return (root->val>leftval)&&(root->val<rightval)&&check(root->left,leftval,root->val)&&check(root->right,root->val,rightval);    }    bool isValidBST(TreeNode* root) {        return check(root,LONG_MIN, LONG_MAX);    }

还有一个细节需要注意下,在leetcode的测试用例中,会把root的值设置为INT_MAX,所以我采用了long。

0 0
原创粉丝点击