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

题目

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

思路

• 二叉搜索树的特性是，左右孩子的值分别小于、大于根节点的值。
• 后序遍历就是先访问左右孩子，再访问根节点。
• 所以后续遍历数组的最后一个值一定是根节点。
• 根节点之前的数组部分，可以分为两个区间，两个区间的元素值分别小于、大于根节点的值，它们代表跟节点的左右子树的元素。
• 这两个区间各自的最后一个元素代表了左右子树的根节点，然后同上。
• 判断一个数组是否为二叉搜索树的后续遍历结果，方法是，先从根节点向前找到第一个小于根节点的数a, 则a之前的元素都为根节点的左子树，看左子树的元素是否都小于根节点值。如果都小于，则继续递归判断左右子树是否满足条件。否则返回返回false。

第二遍思路

• 后序遍历，序列最后一个元素是根节点。最后一个元素前面部分可以分为两部分，一部分是左子树，值全小于最后一个元素；一部分是右子树，值都大于最后一个元素。
• 每次通过根节点的数（序列最后一个数），从后向前，找到第一个小于根节点的数，这个数的左面是左子树，右边是右子树。判断左子树的值是否都小于根节点。不满足返回false;满足遍历左右子树。
• 如果区间剩下的数不多于两个，那么肯定是合理的，直接返回true。

问题

没有想到后序遍历数组的特点。

代码

class Solution {
public:
    bool VerifySquenceOfBST(vector<int> sequence) {
        if ( sequence.size() == 0 ) return false;
        
        return helper( sequence, 0, sequence.size()-1 );
    }
    
    bool helper( vector<int>& sequence, int start, int end ) {
        // 区间只剩单元素。或者区间为空（一个root没有右子树，没有左子树的情况在20行考虑了）
        if ( end - start < 1 ) return true;
        
        int root = sequence[end];
        // 从后向前找到第一个小于root的值。
        int mid = end - 1;
        while ( mid >= start && sequence[mid] > root ) {
            --mid;
        }
        
        if ( mid < start ) { // 如果没有左子树
            return helper( sequence, start, end-1 );
        }
        
        // 判断mid左边的元素是否都小于root。
        for ( int j = start; j <= mid; ++j ) {
            if ( sequence[j] > root )
                return false;
        }
        
        return helper( sequence, start, mid )
            && helper( sequence, mid+1, end-1 );
    }
};

第二遍代码

class Solution {
public:
    bool VerifySquenceOfBST(vector<int> sequence) {
        if ( sequence.size() == 0 ) return false;
        
        return helper( sequence, 0, sequence.size() - 1 );
    }
    
    bool helper( vector<int>& sequence, int lo, int hi ) {
        // 区间元素不大于2
        if ( hi - lo + 1 <= 2 ) return true;
        
        int rootNum = sequence[hi];
        
        int mid = hi - 1;
        while ( mid >= lo && sequence[i] > rootNum )
            --mid;
        
        for ( int i = lo; i < mid; ++i ) {
            if ( sequence[i] > rootNum )
                return false;
        }
        
        return helper( sequence, lo, mid )
            && helper( sequence, mid+1, hi-1 );
    }
};