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

204 阅读1分钟

👉 “Offer 驾到,掘友接招!我正在参与2022春招打卡活动点击查看 活动详情

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

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

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

     5
    / \
   2   6
  / \
 1   3

示例 1:

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

示例 2:

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

限制:

数组长度 <= 1000

解题思路:递归

  • 根据后序遍历的最后一个元素是根结点,划分左右子树
  • 又因为二叉搜索树的左分支都小于根节点,右分支都大于根节点
  • 递归的将其划分如样例[1,3,2,6,5]被划分为[1,3,2] [6] [5]
class Solution {
public:
    bool verifyPostorder(vector<int>& postorder) {
        int n = postorder.size();

        function<bool(int,int)> isOK = [&](int l,int r){
            if(l >= r) return true;
            int i = l, cur = postorder[r];
            while(postorder[i] < cur) i++;
            int j = i;
            while(postorder[j] > cur) j++;
            return j == r and isOK(l,i - 1) and isOK(i, j - 1);
        };

        return isOK(0,n - 1);
    }
};