程序员面试题精选－判断整数序列是不是二元查找树的后序遍历结果

转自何海涛日志http://zhedahht.blog.163.com/

题目：输入一个整数数组，判断该数组是不是某二元查找树的后序遍历的结果。如果是返回true，否则返回false。

例如输入5、7、6、9、11、10、8，由于这一整数序列是如下树的后序遍历结果：

         8
       /  \
      6    10
    / \    / \
   5   7   9  11

因此返回true。

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

分析：这是一道trilogy的笔试题，主要考查对二元查找树的理解。

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

以下为我的实现代码，仅供参考

//判断整数序列是否为二叉查找树的后序遍历序列
#include<iostream>
using namespace std;

bool VerifyPST(int *sequence, int length)
{
	//两个元素之内的序列必然可以是二叉树的后序遍历序列，包括0个元素 
	if(length <= 2)
		return true;
		
	//根元素为序列最后一个元素 
	int root = sequence[length-1];
	
	int i=0; 
	//找到第一个大于根的元素 
	for(; i<length-1; i++)
	{
		if(sequence[i]>root)
			break;
	}
	//判断从第一个大于根的元素 i 到根元素之前一个元素 (length-1) 之间的子序列中元素是否均大于根;
	//若非，则可断定此序列不是二叉树后序遍历序列；若是，则递归判定i左右两侧子序列 
	for(int j=i; j<length-1; j++)
	{
		if(sequence[j]<root)
			return false;
	}
	return VerifyPST(sequence, i) && VerifyPST(sequence+i, length-i-1);
}

bool IsPST(int *sequence, int length)
{
	if(!sequence || length<=0)
		return false;
	return VerifyPST(sequence, length);
} 


int main()
{
	int sequence[] = {5,7,6,9,11,10,8};
	int length = sizeof(sequence)/sizeof(int);
	if(IsPST(sequence, length))
		cout<<"true"<<endl;
	else
		cout<<"false"<<endl;
	system("pause");
	return 0;
}