剑指 Offer 33. 二叉搜索树的后序遍历序列

剑指 Offer 33. 二叉搜索树的后序遍历序列

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

参考以下这颗二叉搜索树：

     5
    / \
   2   6
  / \
 1   3

示例 1：

输入: [1,6,3,2,5]
输出: false

示例 2：

输入: [1,3,2,6,5]
输出: true

---

一开始想的是中序和后续还原二叉树，但是很麻烦，比对条件也比较麻烦,朋友写出来了一版

class Solution
{
public:
    bool verifyHelper(vector<int> &postoder, vector<int> &inorder, int start_post, int end_post, int start_in, int end_in)
    {
        if (start_post > end_post || start_in > end_in)
            return true;
        int index = start_in;
        bool flag = false;
        for (; index <= end_in; index++)
        {
            if (inorder[index] == postoder[end_post])
            {
                flag = true;
                break;
            }
        }
        if (!flag)
            return false;
        int L = index - start_in;
        int R = end_in - index;
        return verifyHelper(postoder, inorder, start_post, start_post + L - 1, start_in, index - 1) &&
               verifyHelper(postoder, inorder, start_post + L, end_post - 1, index + 1, end_in);
    }
    bool verifyPostorder(vector<int> &postorder)
    {
        vector<int> inOrder = postorder;
        sort(inOrder.begin(), inOrder.end());
        return verifyHelper(postorder, inOrder, 0, postorder.size() - 1, 0, inOrder.size() - 1);
    }
};

最后看了题解，脑子越来越不好

class Solution {
public:
    bool verifyPostorder(vector<int>& postorder) {
        return helpe(postorder, 0, postorder.size() - 1);
    }

private:
    bool helpe(vector<int> &postorder, int l, int r) {
        if (l >= r) return true;
        int p = l, m = 0;
        while (postorder[p] < postorder[r]) ++p;
        m = p; // 切割点
        while (postorder[p] > postorder[r]) ++p;
        return p == r && helpe(postorder, l, m - 1) &&
            helpe(postorder, m, r - 1);
    }
};