题目描述:
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同。
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-hou-xu-bian-li-xu-lie-lcof/
解题思路:
- 二叉搜索树的后序遍历序列。则最后的为根节点
- 后序遍历时,左子树的值都比根节点值小
- 当某节点比根节点值大时,则该节点为右子树,即右子树的值均大于根节点的值
- 基于第三点性质,判断该序列是否为二叉搜索树
1.利用bst的性质
代码:
class Solution {
public:
bool verifyPostorder(vector<int>& postorder) {
int sz = postorder.size();
if(sz <= 1)
return true;
int start = 0;//起始位置
int root = 0;//根节点值
int i = 0;
while(start < sz)
{
int pos = sz - 1 - start;//当前所在根节点,从最后一个位置开始计算
root = postorder[pos];//当前所在根节点的值
i = 0;//从头开始寻找左子树大于当前根节点的位置
while(postorder[i] < root)//左子树的值都比根节点值小
++i;
for(;i < pos; ++i)//右子树的值
{
if(postorder[i] < root)//右子树值比根节点小,则不为二叉搜索树
return false;
}
++start;
}
return true;
}
};
2.单调栈
借助一个辅助栈,寻找每一个左子节点的根节点。
- 左子节点的值都小于根节点值,则找到其根节点即可;
- 如果当前值大于其根节点,那么就不是二叉搜索树
- 栈顶元素大于当前节点,循环变换其根节点的值,并且出栈
代码:
class Solution {
public:
bool verifyPostorder(vector<int>& postorder) {
int sz = postorder.size();
if(sz <= 1)
return true;
stack<int> st;
int root = INT_MAX;
for(int i = sz - 1; i >= 0; --i)
{
if(postorder[i] > root)//当前元素大于根节点
return false;
while(!st.empty() && st.top() > postorder[i])//栈不空,并且栈顶元素大于当前节点值
{
root = st.top();
st.pop();
}
st.push(postorder[i]);
}
return true;
}
};