题目
原题链接
给定一个二叉树,判断其是否是一个有效的二叉搜索树。
假设一个二叉搜索树具有如下特征:
- 节点的左子树只包含小于当前节点的数。
- 节点的右子树只包含大于当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
示例 1:
输入:
2
/ \
1 3
输出: true
示例 2:
输入:
5
/ \
1 4
/ \
3 6
输出: false
解释: 输入为: [5,1,4,null,null,3,6]。
根节点的值为 5 ,但是其右子节点值为 4 。
思路
为了验证一棵树是否是BST,我们可以一个节点一个节点的查看。在本人的思路中,每一个节点都有一个最大值和最小值的范围。为了理解这个,我们举个例子。
5
/ \
1 8
/ \
3 10
在上述的树当中,1比5小,8比5大,OK
向下,3比8小,10比8大,OK……OK吗?到底OK不OK?
不OK!因为3在5的右子树,应当比5大。
注意题目中的一句话:
- 节点的左子树只包含小于当前节点的数。
- 节点的右子树只包含大于当前节点的数。
是整棵子树都包含大于(小于)当前节点的数,整棵子树!
所以不可以直观地认为一个节点只要比父节点大或者小就可以了,它实际上有一个限定的大小范围。
对于这个3,它的范围就是(5, 8)。
有人就会问了,说得轻巧,那么程序实现中,最大值和最小值怎么更新呢?
很简单,如果要接下来要检查的节点在这个节点的左边,那么最大值就是这个节点的值,最小值就是上一轮检查当中的最小值,反之亦然。
以下是代码,首先将结点的值与上界和下界(如果有的话)比较。接下来对左子树和右子树递归进行该过程。
/**
* 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) {
return helper(root, NULL, NULL); //这是题目给我们的函数
}
bool helper(TreeNode* root, int* low, int* high) {
if(!root) return true; //空树也符合二叉搜索树的定义
int* val = &root -> val; //将根节点的值赋给val
if(low && *val <= *low) return false;
if(high && *val >= *high) return false;
if(!helper(root -> right, val, high)) return false;
if(!helper(root -> left, low, val)) return false;
return true;
}
};
为什么要用指针?最重要的一点就是在接下来两行中,low和high判断是否为null(也就是上文提到的是否有上界和下界),只有指针能实现。
if(low && *val <= *low) return false;
if(high && *val >= *high) return false;
另外,leetcode还有两个测试点有涉及到INT_MAX和INT_MIN,如果不用指针,很难保证代码不在比对大小是出错(比如是>还是≥,是<还是≤),不信你去试试。
厚颜无耻的我hh