题目大意
给定一个树结构,判定这个树是不是二叉搜索树。
分析
首先给出二叉搜索树的定义:
1.节点的左子树只包含小于当前节点的数。
2.节点的右子树只包含大于当前节点的数。
3.所有左子树和右子树自身必须也是二叉搜索树。
这里有几个要点,小于、大于意味着比较的时候等于不能满足条件,之后要求的是左子树所有节点都要小于当前节点,右子树的所有节点都要大于当前节点,这意味着不仅儿子节点要满足,其后的所有子孙节点都要满足。
那么我们不妨在递归的时候加入small和big两个vector,要求当前节点的值要小于small中的元素,大于big中的元素,当判定左儿子的时候就向small中加入当前节点,然后向下继续判定,右儿子的话则在big中进行类似操作。
代码
bool dfs(TreeNode* root, vector<int> small, vector<int> big) {
if(root == NULL) {
return true;
}
int val = root -> val;
for(int i = 0; i < small.size(); i++) {
if(val >= small[i]) {
return false;
}
}
for(int i = 0; i < big.size(); i++) {
if(val <= big[i]) {
return false;
}
}
bool left = true, right = true;
if(root -> left != NULL) {
small.push_back(val);
left = dfs(root -> left, small, big);
small.pop_back();
}
if(root -> right != NULL) {
big.push_back(val);
right = dfs(root -> right, small, big);
}
return left && right;
}
bool isValidBST(TreeNode* root) {
vector<int> small, big;
small.clear();
big.clear();
return dfs(root, small, big);
}