题目描述
输入一个非空整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
思路1:后序遍历,数组最后一个元素必为根节点。
1、先找到根节点root,位于数组末尾;
2、从左到右遍历数组,得到第一个比root大的元素位置i,将数组分为两部分,i左边为左子树,i开始往右为右子树;
3、判断右子树是否满足>root
4、递归1,2,3
class Solution
{
public:
bool VerifySquenceOfBST(vector<int> sequence) {
if(sequence.size()==0)
return false;
return VerifySquence(sequence, 0, sequence.size() - 1);
}
bool VerifySquence(vector<int> sequence, int l,int r) {
if (l >= r) //递归终止条件
return true;
int root_val = sequence[r];
int i;
for (i = l; i < r; i++) {
if (sequence[i] > root_val)
break;
}
for (int j = i; j < r; j++) {
if (sequence[j] < root_val)
return false;
}
return VerifySquence(sequence, l, i - 1) && VerifySquence(sequence, i, r - 1);
}
};
思路2:二叉树的中序和后序序列满足“栈的压入弹出”,即中序作为栈的压入序列,后续则为该栈的弹出序列;而二叉搜索树BST的中序即为排序数组
class Solution
{
public:
bool VerifySquenceOfBST(vector<int> sequence) {
vector<int> InOrderseq = sequence; //复制
sort(InOrderseq.begin(), InOrderseq.end()); //BST为中序排序序列
return IsPopOrder(InOrderseq, sequence);
}
bool IsPopOrder(vector<int> pushInOrder, vector<int> popLastOrder) {
if (pushInOrder.size() == 0 || popLastOrder.size() == 0)
return false;
stack<int> st;
int popIndex = 0; //记录后序(弹栈)序列的索引值
for (int i = 0; i < pushInOrder.size(); i++) {
st.push(pushInOrder[i]); //将中序序列压入栈中
while (!st.empty() && st.top()==popLastOrder[popIndex]) //比较栈顶元素与弹栈元素是否相等
{
popIndex++;
st.pop();
}
}
return st.empty(); //若为空,说明中序和后序序列满足栈的压入弹出,返回true
}
};