您的位置:首页 > 其它

leetcode:Validate Binary Search Tree

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

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。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: