【剑指offer】23.二叉搜索树的后序遍历序列

题目

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

---

分析

二叉搜索树的性质：根节点大于左子树所有元素，小于右子数的所有元素。那么后序遍历的话，最后一个元素就为根节点root。那么本题的思路如下：

1. 找到数组中左右子树的分界点，并判断右子树是否全大于根节点，若不是则直接返回false；
2. 之后递归判断左右子树是否都为二叉搜索树，注意左子树或右子树不存在的情况；
3. 左右子树都为二叉搜索树返回true，左右子树有一个不为二叉搜索树则返回false；

github链接：JZ23-二叉搜索树的后序遍历序列

---

C++代码

#include <iostream>
#include <vector>
using namespace std;

 
class Solution {
	public:
	    bool VerifySquenceOfBST(vector<int> sequence) {
	    	int size = sequence.size();
	    	return this->veriftBST(sequence,0,size-1);
	    }
	    
	    bool veriftBST(vector<int> sequence , int start, int end){
	    	if(sequence.empty() || start > end){
	    		return false;
			}
			
			int root = sequence[end];
			//找到左右子树的分界点
			int index = start; 
			for(index = start ; index < end ; index++){
				if(sequence[index] > root){
					break;
				}
			}
			//index为右子树的第一个元素下标 
			for(int i = index ; i < end ; i++){
				if(sequence[i] < root){
					return false;
				}
			}
			bool left = true;
			if(index > start){		//有左子树 
				left = this->veriftBST(sequence,start,index-1);
			}
			bool right = true;
			if(end-1 > index){
				right = this->veriftBST(sequence,index,end-1);
			}
			return left && right;
		}
};

int main()
{
	int n;
	while(cin>>n){
		vector<int> array(n);
		for(int i = 0 ; i < n ; i++){
			cin>>array[i];
		}
		Solution s;
		cout<<s.VerifySquenceOfBST(array);
	}
	
	return 0;
}