剑指 Offer -- 二叉搜索树的后序遍历序列(二十三)

二叉搜索树的后序遍历序列(二十三)

题目描述

输入一个整数数组，判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。

代码(已在牛客上 AC)

使用 imin 和 imax 来判断节点中的值满足的范围. 如果数组是 BST 的后序遍历, 那么需要找到左子树和右子树的分界点, 方法就是使用 lower_bound, 因为左子树中的值都小于 nums[j], 右子树中的值都大于 nums[j], lower_bound 可以找到第一个属于右子树的节点.

判断条件 idx == j 是说, lower_bound 返回的是 end()(注意, 由于 lower_bound 中考察 nums 在 [i ... j] 范围内的元素, 而 nums[j] 是根节点, 所以 idx == j 说明找不到第一个大于或等于 nums[j] 的值, 说明只有左子树.

否则, 判断两棵子树是否都符合 BST 的要求.

class Solution {
public:
    bool VerifySquenceOfBST(vector<int> seq) {
        if (seq.empty()) return false;
        return verify(seq, 0, seq.size() - 1, INT32_MIN, INT32_MAX);
    }
private:
    bool verify(vector<int> &nums, int i, int j, int imin, int imax) {
        if (i > j) return true;
        if (i == j && !(imin < nums[j] && nums[j] < imax)) return false;
        int idx = std::lower_bound(nums.begin() + i, nums.begin() + j + 1, nums[j]) - nums.begin();
        if (idx == j) return verify(nums, i, j - 1, imin, nums[j]);
        return verify(nums, i, idx - 1, imin, nums[j]) &&
            verify(nums, idx, j - 1, nums[j], imax);
    }
};

我早期的写法, 自己实现一个 BinarySearch.

class Solution {
public:
    bool VerifySquenceOfBST(vector<int> sequence) {
        if (sequence.empty()) return false;
        return VerifySquenceOfBST(sequence, 0, sequence.size() - 1, INT32_MIN, INT32_MAX);
    }
private:
    bool VerifySquenceOfBST(vector<int> &nums, int i, int j, int imin, int imax) {
        if (i > j) return true;
        if (!(imin < nums[j] && nums[j] < imax)) return false;
        int idx = binarySearch(nums, i, j - 1, nums[j]);
        if (idx == -1) return VerifySquenceOfBST(nums, i, j - 1, nums[j], imax);
        return VerifySquenceOfBST(nums, i, idx, imin, nums[j]) &&
            VerifySquenceOfBST(nums, idx + 1, j - 1, nums[j], imax);
    }
    // binarySearch 这里找到最后一个属于左子树的节点的索引
    int binarySearch(vector<int> &nums, int l, int r, int target) {
        if (l > r) return -1;
        int start = l;
        while (l <= r) {
            int mid = l + (r - l) / 2;
            if (nums[mid] < target) l = mid + 1;
            else r = mid - 1;
        }
        if (r < start) return -1;
        return l - 1;
    }
    // search 是 O(n) 的
    int search(vector<int> &nums, int l, int r, int target) {
        for (int i = r; i >= l; --i) {
            if (nums[i] < target) return i;
        }
        return -1;
    }
};