题目描述:
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则返回true,否则返回false。假设输入的数组的任意两个数字都互不相同。
示例1
输入
[4,8,6,12,16,14,10]
返回值
true
在线链接
代码:
import java.util.*;
public class Solution {
public boolean VerifySquenceOfBST(int[] sequence) {
if (sequence == null || sequence.length == 0) {
return false;
}
Stack<Integer> st = new Stack<>();
int parent = Integer.MAX_VALUE;
for (int i = sequence.length - 1; i >= 0; i--) {
if (parent < sequence[i]) {
return false;
}
while (!st.isEmpty() && st.peek() > sequence[i]) {
parent = st.pop();
}
st.add(sequence[i]);
}
return true;
}
}
思路概述:
例子中的后序遍历数组:
[4,8,6,12,16,14,10],对应的二叉树
我们将这个数组倒过来看
10 14 16 12 6 8 4
倒序的结点顺序为 左、右、根,将倒序遍历倒置后
变成了 根、右、左,而且我们知道,二叉搜索树有一个性质,就是结点值大小为
左<根<右。
所以我们假设倒置后的倒序数组为nums,如果nums[i] < nums[i+1],那么nums[i+1]肯定是nums[i]对应结点的右结点,因为挨着且大小对应。
如果nums[i] > nums[i+1],那么nums[i+1]肯定是前置位的一个子结点。比如例子中,10 14 16,是升序,可以确定是这样子
12肯定是三个结点的一个子结点,按照二叉搜索树的性质,12不可能是10的左节点(因为12>10),也不可能是16的左结点(因为12<14),所以12是14的左结点。也就是说:如果nums[i] < nums[i+1],那么nums[i+1]肯定是前置位大于该结点的最小值的结点。找到了14的左子树结点,那么之后的元素就不可能再比14大(因为顺序根、右、左)
同理,接下来的6,大于他的最小值为10,那就是10的左结点…
代码讲解:
倒着遍历数组,用一个栈记录,如果栈不为空,且栈顶元素>数组当前元素,就出栈,并将出栈元素赋值给parent,找到第一个小于当前元素的结点值,上一个出栈的就是其父结点,值刚刚赋给了parent。所以我们每次遍历,先判断parent是否大于当前元素即可。