题目描述
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
解题思路
递归解法
可以将输入序列划分为3部分,即left、right、root,首先找到left部分最后一个结点的下标,即可完成分隔。如果left部分和right部分均是BST,即可递归调用VerifySquenceOfBST( )函数,变量bleft记录left部分是否为BST,bright记录right部分是否为BST。i从0~len-1对所有结点遍历一次。最后的 bleft&&bright 即为所求的值。
class Solution {
public:
bool VerifySquenceOfBST(vector<int> sequence) {
//递归方法
int len=sequence.size();
if(len<=0)
return false;
vector<int> left, right;
int root=sequence[len-1];
int i=0;
while(i<len-1) // 处理left部分
{
if(sequence[i]>root) break;
left.push_back(sequence[i]);
i++;
}
int j=i; // 处理right部分,此时i为left部分最后一个结点的下标
while(j<len-1)
{
if(sequence[j]<root) return false;
right.push_back(sequence[j]);
j++;
}
bool bleft=true; // 应初始化为true,left部分是BST序列,才能调用VerifySquenceOfBST()
// i!=0表示有左子树
if(i != 0) bleft=VerifySquenceOfBST(left); // i为left部分最后一个结点的下标
bool bright=true;
// i!=len-1表示有右子树
if(i != len-1) bright=VerifySquenceOfBST(right);
return (bleft && bright);
}
};
非递归解法
非递归也是一个基于递归的思想:
左子树一定比右子树小,因此去掉根后,数字分为left,right两部分,right部分的最后一个数字是右子树的根,它也比左子树所有值大,因此我们可以每次只看有子树是否符合条件即可,即使到达了左子树,左子树也可以看出由左右子树组成的树还想右子树那样处理对于左子树回到了原问题,对于右子树,左子树的所有值都比右子树的根小,可以暂时把它看出右子树的左子树,只需看看右子树的右子树是否符合要求即可。
class Solution {
public:
bool VerifySquenceOfBST(vector<int> sequence) {
//非递归方法
int i = 0;
int size = sequence.size(); //end指向最后一个结点,即根结点
if(size == 0)
return false;
while(--size){
while(sequence[i++]<sequence[size]); //left部分
while(sequence[i++]>sequence[size]); //right部分
if(i<size)
return false;
i = 0;
}
return true;
}
};