题目:输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相空间
思路:
由于题目没有给出二叉树长什么样子,只给了一个数组,所以我们要从数组中找到二叉搜索树后序序列的规律。
后序序列的最后一个元素是树的根结点,因此,可以再数组中找到一个划分,根结点的左子树的值都比根结点小,右子树的值都比根结点大,用这个思路(左大右小)就可以不断递归判断子树是不是二叉搜索树了。
- 时间复杂度O(n^2)
- 空间复杂度O(n)
class Solution {
public boolean verifyPostorder(int[] postorder) {
return checkSub(postorder, 0, postorder.length - 1);
}
public static boolean checkSub(int[] postorder, int start, int end) {
// 只有一个元素或者没有元素,是二叉搜索树
if (start >= end) {
return true;
}
int pivot = postorder[end]; // pivot 记录最后一个元素
int pivotIndex = start; // pivotIndex 用于记录分界线的下标
// 找到第一个大于 pivot 的数的下标,等号保留,可能是单边树
while (pivotIndex <= end && postorder[pivotIndex] < pivot) {
pivotIndex++;
}
for (int i = pivotIndex; i <= end; i++) {
// 根据左小右大的规律,如果 pivotIndex 后面的数有小于 pivot 的,直接返回false
if (postorder[i] < pivot) {
return false;
}
}
return checkSub(postorder, start, pivotIndex - 1) && checkSub(postorder, pivotIndex, end - 1);
}
}
同样的方法,网友写的代码看起来也太舒服了
class Solution {
public boolean verifyPostorder(int[] postorder) {
return recur(postorder, 0, postorder.length - 1);
}
public static boolean recur(int[] postorder, int start, int end) {
if (start >= end) {
return true;
}
int pivotIndex = start;
while (postorder[pivotIndex] < postorder[end]) {
pivotIndex++;
}
int temp = pivotIndex;
while (postorder[temp] > postorder[end]) {
temp++;
}
return temp == end && recur(postorder, start, pivotIndex - 1) && recur(postorder, pivotIndex, end - 1);
}
}