题目描述:输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true
,否则返回 false
。假设输入的数组的任意两个数字都互不相同。
参考以下这颗二叉搜索树:
5
/ \
2 6
/ \
1 3
示例 1:
输入: [1,6,3,2,5]
输出: false
示例 2:
输入: [1,3,2,6,5]
输出: true
题目解析
二叉搜索树定义:根节点的左子树中的值均小于根节点的值,右子树的值均大于根节点的值。
后序遍历定义:左子树、右子树、根
根据上面两个定义可知,根节点是数组postorder的最后一个节点,如果此数组代表的是二叉搜索树,则postorder的剩下值中必会左半部分为会小于根节点的值,右半部分大于根节点的值。否则不是二叉搜索树后序遍历的结果。此步骤使用递归来实现。
递归解析:
初始条件:设l=0, r=len-1,其中 len是数组的长度。
终止条件: 当 l≥r,说明此子树节点数量 ≤1,无需判别正确性,因此直接返回 true。
递推工作:
1. 划分左右子树:
遍历后序遍历的 [l,r] 区间元素,寻找第一个大于根节点的节点,索引记为middle。此时,可划分出左子树区间 [l, middle−1] 、右子树区间 [middle, r−1] 、根节点索引 r 。
2. 判断是否为二叉搜索树:
左子树区间 [l, middle−1] 内的所有节点都应<postorder[r] 。而划分左右子树的步骤已经保证左子树区间的正确性,因此只需要判断右子树区间即可。右子树区间 [middle, r−1] 内的所有节点都应 >postorder[r] 。实现方式为遍历。当遇到 ≤postorder[r] 的节点则跳出,通过 index 是否等于 r 来判断此数组是否为为二叉搜索树的后序遍历。
class Solution {
public:
bool verifyPostorder(vector<int>& postorder) {
return verify(postorder, 0, postorder.size()-1);
}
bool verify(vector<int>& postorder, int l, int r){
if(l>=r){
return true;
}
int index=l;
while(postorder[index]<postorder[r]){
index++;
}
int middle=index;
while(postorder[index]>postorder[r]){
index++;
}
return index==r&&verify(postorder, l, middle-1)&&verify(postorder, middle, r-1);
}
};