[剑指offer]二叉搜索树的后序遍历序列

二叉搜索树的后序遍历序列

题目

输入一个整数数组，判断该数组是不是某二叉搜索树的后序遍历的结果。

如果是则返回true，否则返回false。

假设输入的数组的任意两个数字都互不相同。

样例

输入：[4, 8, 6, 12, 16, 14, 10]

输出：true

题解

以二叉搜索树根节点为界，根节点左边的必然都小于根节点，根节点右边的必然都大于根节点，如果违背该原则，则说明该数组不是某二叉搜索树的后序遍历的结果

class Solution {
public:
    vector<int> sq;
    
    bool dfs(int l, int r) {
        if (l > r) return true;
        int k = l, root = sq[r];
        while (k < r && sq[k] < root) k ++;
        for (int i = k; i < r; i ++)
            if (sq[i] < root)
                return false;
        return dfs(l, k-1) & dfs(k, r-1);
    }
    
    bool verifySequenceOfBST(vector<int> sequence) {
        sq = sequence;
        return dfs(0, sq.size() - 1);
    }
};