【剑指Offer系列33】二叉搜索树的后序遍历序列

题目

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

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

     5
    / \
   2   6
  / \
 1   3

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

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

提示：
数组长度 <= 1000

代码

Python

# 思路：
# 抓住二叉搜索树规律：左子树所有节点小于根节点，右子树所有节点大于右节点；采用递归实现
# 复杂度：
# O(N^2)
class Solution:
    def verifyPostorder(self, postorder: List[int]) -> bool:
        def recur(i,j):
            if i>=j: return True
            m=i
            while postorder[m]<postorder[j]: m+=1 # 左子树
            n=m
            while postorder[m]>postorder[j]: m+=1 # 右子树
            return m==j and recur(i,n-1) and recur(n,j-1)
        return recur(0,len(postorder)-1)

C++

class Solution {
public:
    bool verifyPostorder(vector<int>& postorder) {
        return recur(postorder,0,postorder.size()-1);
    }
    
    bool recur(vector<int>& postorder,int i,int j) {
        if (i>=j) return true;
        int m=i;
        while (postorder[m]<postorder[j]) m++;
        int n=m;
        while (postorder[m]>postorder[j]) m++;
        return (m==j&&recur(postorder,i,n-1)&&recur(postorder,n,j-1));
    }
};