LeetCode 98 验证二叉搜索树 题解

题目大意

给定一个树结构，判定这个树是不是二叉搜索树。

分析

首先给出二叉搜索树的定义：

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);
}