题目描述
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同。
示例1:
输入: [1,6,3,2,5]
输出: false
示例2:
输入: [1,3,2,6,5]
输出: true
问题分析
递归方法
要点:
1、后序遍历的最后一个必为根结点。即左/右/根
2、根结点将遍历结果可以分成两部分。即左子树,柚子树
3、左子树的元素值都小于根结点,可以以此来切割子区间。
4、切割后的右区间如果有元素的值小于根结点,则该遍历顺序肯定不对
具体代码实现如下所示:
注意区间的表示,这里统一为左闭右开区间
class Solution {
private:
//左闭右开区间
bool dfs(vector<int>& postorder, int begin, int end){
if(begin >= end) return true;
int rootval = postorder[end - 1];
int mid = begin;
//找到第一个比根结点大的节点mid,此时从0到mid(左闭右开)都为左子树
while(postorder[mid] < rootval){
mid++;
}
//需要遍历查看mid到end-1(左闭右开区间)里的元素是不是都大于end-1(rootroot)节点的元素
int temp = mid;
while(temp < end - 1){
if(postorder[temp] < rootval){
return false;
}
temp++;
}
//二分的思路,分别递归左右子树,注意区间都是左闭右开
return dfs(postorder, begin, mid) && dfs(postorder, mid, end - 1);
}
public:
bool verifyPostorder(vector<int>& postorder) {
return dfs(postorder, 0, postorder.size());
}
};
stack栈的方式
要点:
对于[1,3,2,6,5](左,右,根),如果反序,将变为[5,6,2,3,1](根,右,左)。
这时,采用栈来辅助操作:
1、初始时栈为空,遍历数组,将一个元素入栈,(对于正确的后序遍历的反序,第一个入栈的元素为根结点,如5)
2、如果当前遍历的元素大于栈顶元素,则将该元素入栈(该元素为根结点的右子树的节点)
3、如果当前遍历的元素小于栈顶元素,则弹出栈顶元素,直到栈顶元素小于该元素,或者栈空。在此过程中,弹出的栈顶元素都有可能是当前所遍历的元素的根结点(根结点大于左子树的值),但是只有大于该元素的最小的值才为根结点(排除右子树节点的影响)。所以要不断更新根结点的值
4、如果栈空后,更新到的根结点的值小于当前遍历的结点,说明该“后序遍历”错误,返回false;
5、不是情况4,就将该元素入栈
class Solution {
public:
bool verifyPostorder(vector<int>& postorder) {
stack<int> stk;
int root = INT_MAX;
//[1,3,2,6,5] 左/右/根
//倒序[5,6,2,3,1] 根/右/左
for(int i = postorder.size() - 1; i >= 0; i--){
int cur = postorder[i];
//stk不是空,当前遍历到的元素小于栈顶元素,说明该元素是栈中元素的左节点
//此时弹出栈内的元素,并更新该子树的根结点的值,找到栈内大于当前遍历到的元素的最小元素作为新的根结点
while(!stk.empty() && stk.top() > cur){
root = stk.top();
stk.pop();
}
//如果栈弹空,更新后的根结点小于当前遍历到的元素(左子树上的点),则返回false,不成立
if(cur > root){
return false;
}
//更新栈或者将此左子树的点放进栈中
stk.push(cur);
}
return true;
}
};```