题目
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同。
参考以下这颗二叉搜索树:
5
/ \
2 6
/ \
1 3
示例 1:
输入: [1,6,3,2,5]
输出: false
示例 2:
输入: [1,3,2,6,5]
输出: true
提示:
数组长度 <= 1000
题解
思路:
二叉搜索树的定义:左子树节点值<根节点值<右子树节点值
在后续遍历中,根节点始终在数组最后一位,那么如何找到二叉搜索树根节点的左节点集合和右节点集合呢?
只需要遍历数组,直到找到一个数>root根节点,说明这个数往左(不包括这个数)都是左子节点集合,这个数往右(包括这个数)就是右子节点集合。
找到了左子节点和右子节点集合,接下来就是递归判断两个子节点集合是否也满足二叉搜索树的定义。
代码
class Solution {
public:
bool verifyPostorder(vector<int>& postorder) {
if(postorder.empty()) return true;//递归终止条件
int root = postorder.back();//获取后续遍历数组的根元素(最后一位)
vector<int> left;//左子节点集合树
vector<int> right;//右子节点集合树
int i,j;
for(i=0;i<postorder.size()-1;i++){//从左往右找比root小的元素,放进left数组
if(postorder[i]<root)
left.push_back(postorder[i]);
else //但凡遇到比root大的元素,跳出循环
break;
}
for(j=i;j<postorder.size()-1;j++){//j从i开始,往右找比root大的元素,放进right数组
if(postorder[j]>root)
right.push_back(postorder[j]);
else
break;//但凡遇到比root小的元素,跳出循环
}
if(j==postorder.size()-1)//如果此时j刚好是当前root的下标,说明这一层是满足二叉搜索树的定义的,递归判断左右子树是否满足
return verifyPostorder(left)&&verifyPostorder(right);
else //如果此时j不是当前root下标,说明这一层的postorder数组并不是严格的左小右大,直接返回false
return false;
}
};
关注公众号:【THU小鱼干杂货铺】
不间断分享大厂算法题解、实习面经、学习资料等