33. 二叉树的后续遍历
1 题目描述
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true
,否则返回 false
。假设输入的数组的任意两个数字都互不相同。
参考下面这颗二叉搜索树:
5
/ \
2 6
/ \
1 3
输入: [1,3,2,6,5]
输出: true
输入: [1,6,3,2,5]
输出: false
2 题目分析
思路:根据题意,容易看出数组的最后一个元素就是根节点,二叉搜索树的定义又可知左边的小于根节点,
右边的大于根节点。因此数组中前半部分小于等于end,后半部分大于等于end,找到分割点进行如下判断:
1. 找到第一个大于根节点的位置m,从该点开始到根节点前一个位置依次判断该元素之后的值是否都大于根节点;
2. 递归判断左边left ~ m-1
3. 递归判断右边m + 1 ~ end - 1
注意:递归出口是left >= end
3 代码
public boolean verifyPostorder(int[] postorder) {
if (postorder == null || postorder.length <= 2) return true;
int N = postorder.length;
return dfs(postorder, 0, N - 1);
}
private boolean dfs(int[] postorder, int left, int end) {
if (left >= end) return true;
int i = left;
while (postorder[i] < postorder[end]) i++;
int m = i;
// 判断m到end的元素是否都大于end
while (postorder[i] > postorder[end]) i++;
// 递归判断
return i == end && dfs(postorder, left, m - 1) && dfs(postorder, m, end - 1);
}