剑指Offer 1367 判断整数序列是不是二元查找树的后序遍历结果

九度： http://ac.jobdu.com/problem.php?pid=1367

9. 判断整数序列是不是二元查找树的后序遍历结果

题目：输入一个整数数组，判断该数组是不是某二元查找树的后序遍历的结果。
如果是返回true，否则返回false。
例如输入5、7、6、9、11、10、8，由于这一整数序列是如下树的后序遍历结果：
  8
  / \
  6 10
  / \ / \
  5 7 9 11
因此返回true。

如果输入7、4、6、5，没有哪棵树的后序遍历的结果是这个序列，因此返回false。

在后续遍历得到的序列中，最后一个元素为树的根结点。从头开始扫描这个序列，比根结点小的元素都应该位于序列的左半部分；从第一个大于跟结点开始到跟结点前面的一个元素为止，所有元素都应该大于跟结点，因为这部分元素对应的是树的右子树。根据这样的划分，把序列划分为左右两部分，我们递归地确认序列的左、右两部分是不是都是二元查找树。

#include <cstdlib>
#include <iostream>
using namespace std;
#define max 15000
 
bool check(int a[], int begin, int end)
{
     if(begin > end)
              return true;
     int mid = end-1;
     int root = a[end];
     while(a[mid] > root && mid >= begin)
                mid--;
      
     for(int i = begin; i <= mid; i++)
             if(a[i] > root)
                     return false;
     bool left = check(a, begin, mid);
     bool right = check(a, mid+1, end-1);
     return left && right;
}
 
 
int main()
{
    int n;
    int a[max];
    while(cin >> n)
    {
        for(int i = 0; i < n; i++)
                cin >> a[i];
        if(check(a, 0, n-1))
                    cout << "Yes" << endl;
        else
            cout << "No" << endl;
    }
   // system("pause");
    return 0;
}