【LeetCode-树】二叉搜索树的后序遍历序列

题目描述

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

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

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

说明:

  • 数组长度 <= 1000

题目链接: https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-hou-xu-bian-li-xu-lie-lcof/

思路

二叉搜索树的后序遍历序列的最后一个节点是根节点,除去最后一个节点,序列可以分为两个连续的部分:一部分小于根节点的值(左子树),另一部分大于根节点的值(右子树)。所以,我们根据根节点判断第一段序列是不是全都小于根节点的值,第二段序列是不是全都大于根节点的值。如果是的话,递归判断两段序列;否则,返回 false。

代码如下:

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

    bool doVerify(vector<int>& postorder, int left, int right){
        if(left>=right) return true;

        int i = left;
        while(postorder[i]<postorder[right]) i++;
        i--;  // 别忘了 i--

        int j = i+1;
        while(postorder[j]>postorder[right]) j++;

        if(j!=right) return false;
        return doVerify(postorder, left, i) && doVerify(postorder, i+1, right-1);
    }
};

这样写也行(基本没啥区别):

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

    bool judge(vector<int>& postorder, int left, int right){
        if(left>=right) return true;

        int root = postorder[right];
        int i = left;
        while(i<right && postorder[i]<root) i++;
        int j = i;
        while(j<right && postorder[j]>root) j++;
        if(j==right){
            return judge(postorder, left, i-1) && judge(postorder, i, right-1);
        }else return false;
    }
};
posted @ 2020-08-05 21:24  Flix  阅读(216)  评论(0编辑  收藏  举报