题目:输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
思路:二叉搜索树的后序遍历结果的特点:序列的最后为根节点,序列的前半部分为根节点的左子树,所有元素小于根节点,后半部分为右子树,所有元素大于根节点。因此解题步骤如下:
1)确定根节点
2) 从左往右遍历序列,直到发现第一个大于根节点的元素x。此元素之前为左子树。
3)从元素x开始遍历序列,如果在到达根节点之前发现有小于根节点的元素,则判定该序列不是二叉搜索树的后序遍历结果,返回false。
4)若上步未返回false,则递归检查左子树序列和右子树序列是否为二叉搜索树的后序遍历结果。
代码(已在牛客网AC):
public boolean VerifySquenceOfBST(int [] sequence) {
if(sequence == null || sequence.length == 0)
return false;
return VerifySquenceOfBST(sequence, 0, sequence.length-1);
}
private boolean VerifySquenceOfBST(int[] sequence, int lo, int hi){
//注意边界条件设为lo == hi,可能会导致无限递归调用,栈溢出。
if(lo >= hi)
return true;
int root = sequence[hi];
int rightBegin;
// 确定右子树开始的位置
for( rightBegin = lo ; rightBegin < hi; rightBegin++){
if(sequence[rightBegin] > root)
break;
}
//若右子树中有小于根节点元素,则返回false
for(int i = rightBegin; i < hi; i++)
{
if(sequence[i] < root)
return false;
}
return VerifySquenceOfBST(sequence, lo, rightBegin-1) && VerifySquenceOfBST(sequence, rightBegin, hi-1);
}