题目来自LeetCode,链接:面试题33. 二叉搜索树的后序遍历序列。具体描述为:输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同。
参考以下这颗二叉搜索树:
5
/ \
2 6
/ \
1 3
示例1:
输入: [1,6,3,2,5]
输出: false
示例2:
输入: [1,3,2,6,5]
输出: true
这道题的话我们先假设给定的输入序列就是一个后序遍历序列,然后检查是否合法。因为后序遍历的顺序就是左子节点->右子节点->根节点
,所以我们可以知道序列的最后一个值就是根节点的值。然后我们还需要知道前面的从哪里开始是右子树的节点,因为二叉搜索树的特点,所以我们可以知道,只要从左往右遍历序列找到第一个大于根节点值的,那么从这里往后的必然都是右子树的节点,然后还需要判断这个右子树是否合法(左子树已经是合法的了),其实就是判断这些右子节点的值是否都大于根节点。时间复杂度为
O
(
n
2
)
O(n^{2})
O(n2),空间复杂度为
O
(
n
)
O(n)
O(n)(二叉树退化为单链表的最坏情况下需要的递归栈)。
JAVA版代码如下:
class Solution {
private boolean isValid(int[] postorder, int left, int right) {
if (left >= right) {
return true;
}
int root = postorder[right];
int idx;
for (idx = left; idx < right; ++idx) {
if (postorder[idx] > root) {
break;
}
}
for (int i = idx; i < right; ++i) {
if (postorder[i] < root) {
return false;
}
}
return isValid(postorder, left, idx - 1) && isValid(postorder, idx, right - 1);
}
public boolean verifyPostorder(int[] postorder) {
return isValid(postorder, 0, postorder.length - 1);
}
}
提交结果如下:
Python版代码如下:
class Solution:
def verifyPostorder(self, postorder: List[int]) -> bool:
def isValid(left, right):
if left >= right:
return True
i = left
while i < right:
if postorder[i] > postorder[right]:
break
i += 1
j = i
while j < right:
if postorder[j] < postorder[right]:
return False
j += 1
return isValid(left, i - 1) and isValid(i, right - 1)
return isValid(0, len(postorder) - 1)
提交结果如下: