题目
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
思路
根据二叉搜索树的性质,根节点的左子树都比根节点小,右子树都比根节点大,并且后序序列的最后一个元素为根节点,并且前一部分是左子树,后一部分是右子树。
从头到尾扫描序列,遇到的第一个比根节点大的元素就是右子树的第一个元素,对于左子树进行递归判断即可,对于右子树,如果是BST的后序序列的话,在右子树的所有元素都应该要比根节点大,所以扫描从右子树第一个节点到根节点之间的序列,如果有元素小于根节点,就不是二叉搜索树的后序遍历序列,如果没有问题,再递归判断。
参考代码
class Solution {
public:
bool VerifySquenceOfBST(vector<int> sequence) {
if(sequence.size() == 0) return false;
return judge(sequence, 0, sequence.size() - 1);
}
private:
bool judge(vector<int>& v, int left, int right){
if(left >= right) return true;
int i, j;
for(i = left; i < right; ++i){
if(v[i] > v[right]) break;
}
bool judgeLeft = judge(v, left, i - 1);
if(!judgeLeft || i == right) return judgeLeft;
for(j = i; j < right; ++j){
if(v[j] < v[right]) return false;
}
return judge(v, i, right - 1);
}
};