@author: sdubrz
@date: 8/2/2020 10:46:58 AM
难度: 中等
考察内容: 二叉搜索树
@e-mail: lwyz521604#163.com
题目来自《剑指offer》 电子工业出版社
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同。
参考以下这颗二叉搜索树:
5
/ \
2 6
/ \
1 3
示例 1:
输入: [1,6,3,2,5]
输出: false
示例 2:
输入: [1,3,2,6,5]
输出: true
提示:
- 数组长度 <= 1000
通过次数25,327 提交次数48,585
解法
二叉搜索树的后序遍历数组中可以分为三部分:
- 第一部分位于最左侧,是二叉搜索树根节点的左子树的元素,所有的元素都小于根节点。该部分可能为空。
- 第二部分位于第一部分的左侧,是二叉搜索树根节点的右子树中的所有元素,均大于根节点。该部分可能为空。
- 第三部分是最后一个元素,也就是二叉搜索树的根节点元素。
所以,这道题类似于快速排序中的思想,只需要判断给定的数组中,所有比根节点小的元素均位于比根节点大的元素的左侧就可以了。用递归的思想来做,时间复杂度为 O(nlog(n))
。需要注意的是,当数组中的元素不足3个的时候,一定是某个二叉搜索树的后序遍历。以下是具体的Java程序实现:
class Solution {
public boolean verifyPostorder(int[] postorder) {
int n = postorder.length;
if(n<3) {
return true;
}
return verifyPostorder(postorder, 0, n-1);
}
private boolean verifyPostorder(int[] postorder, int head, int tail) {
if(tail-head<2) {
return true;
}
int ptr1 = head-1;
int ptr2 = tail;
while(ptr1+1<tail && postorder[ptr1+1]<postorder[tail]) {
ptr1++;
}
while(ptr2-1>=head && postorder[ptr2-1]>postorder[tail]) {
ptr2--;
}
if(ptr1==ptr2-1) {
return verifyPostorder(postorder, head, ptr1) && verifyPostorder(postorder, ptr2, tail-1);
}else {
return false;
}
}
}
在 LeetCode 系统中提交的结果为
执行结果:通过 显示详情
执行用时:0 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗:37.2 MB, 在所有 Java 提交中击败了34.81%的用户