题目描述
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
解题思路
回顾后序遍历序列的特性:最后一个元素为根,跟前面的序列可以分为小于根的前半部分(左子树)和大于根的后半部分(右子树)。
因此用递归进行判断是最便利的。开辟两个新vector分别存储小于根的序列和大于根的序列。再分别将两个序列递归判断是否为后序遍历。
结束递归的条件:
分段的后半部分出现比根小的元素,则不是后序,返回false;
分段的前后段都为空,则该序列已完成判断,返回true;
代码
class Solution {
public:
bool VerifySquenceOfBST(vector<int> sequence) {
if(sequence.empty()) return false;
vector<int> left;
vector<int> right;
int i, size=sequence.size();
for(i=0; i<size-1; i++){
if(sequence[i]<sequence[size-1])
left.push_back(sequence[i]);
else break;
}
for(i; i<size-1; i++){
if(sequence[i]<sequence[size-1]) return false;
right.push_back(sequence[i]);
}
if(left.empty() && !right.empty()) return VerifySquenceOfBST(right);
if(!left.empty() && right.empty()) return VerifySquenceOfBST(left);
if(left.empty() && right.empty()) return true;
return VerifySquenceOfBST(left)&& VerifySquenceOfBST(right);
}
};