题目描述
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
知识点:
1、后序遍历是”左右根“。
因此最后一个遍历到根节点,故数组中最后一个元素是根节点
2、二叉搜索树的特性:左子树上的节点都小于根节点,右子树上的节点都大于根节点。
这个后序遍历得到的数组是{1,3,4,2,7,6,5};
可以从数组中判断,前面部分比root小的属于左子树,后面部分为右子树,若在右子树中出现比root小的数,那就不符合二叉搜索树的特性 root=5, 数字1,3,4,2都小于5,位于左侧子树,7,6大于5,位于右侧子树。
注意的是:从第一个大于根节点的数开始,后面的元素都是属于右侧子树的。若右侧子树中出现比根节点大的数,这就不是二叉搜索树。
比如数组{7,4,6,5} 根节点是5,根据“左右根”的遍历顺序和二叉树的特性,7比5大,那么7应该是属于右侧子树的,那7,4,6都是右侧子树。
常规解法:
class Solution {
public:
bool VerifySquenceOfBST(vector<int> sequence) {
return bst(sequence,0,sequence.size()-1);
}
bool bst(vector<int>sequence,int begin,int end)
{
if(sequence.empty()||begin>end)
return false;
// 后序遍历最后一个数是根节点
int root=sequence[end];
// 根据根节点将数组分成左右子树
int i=begin;
for(;i<end;i++)
{
if(sequence[i]>root) //此时i是属于右子树的第一个节点
break;
}
//右边出现比根节点小的数,那这就不是二叉搜索树
for(int j=i;j<end;j++)
{
if(sequence[j]<root)
return false;
}
bool left=true;
if(i>begin) //左子树存在
left=bst(sequence,begin,i-1);
bool right=true;
if(i<end-1)//右子树存在,end已经是当前的根节点了,end-1是下一个的根节点
right=bst(sequence,i,end-1);
return left&&right;
}
};
还有一种解法:
class Solution {
public:
bool VerifySquenceOfBST(vector<int> sequence) {
if(sequence.empty())
return false;
int len=sequence.size();
int i=0;
while(len)
{
int root=sequence[len-1]; //根节点
//计数比root小的前部分数据,直到比root大的数停止,即遇到右子树节点
while(sequence[i]<root)
i++;
//后部分比root大的计数,如果出现了比root小的数,i计数停止
while(sequence[i]>root)
i++;
if(i<len-1) //说明不满足后部分全大于root
return false;
i=0; //注意每次要更新i的值,每次遍历都从头开始
len--; //
}
return true;
}
};