【Lintcode】1307. Verify Preorder Sequence in Binary Search Tree

题目地址：

https://www.lintcode.com/problem/verify-preorder-sequence-in-binary-search-tree/description

给定一个长$n$的数组$A$，判断其是否可能是一个二叉搜索树的前序遍历。题目保证$A$里没有重复数字。

递归写法参考https://blog.csdn.net/qq_46105170/article/details/109635335。下面给出一个栈的写法：

模仿栈模拟先序遍历的过程，遍历数组，当遍历到$A[i]$的时候，如果栈空或者栈顶大于$A[i]$，则将该数字入栈，这对应着先序遍历里的向左走；如果遇到了小于栈顶的$A[i]$，那就说明该数是在某个节点的右子树里，则pop栈顶，直到栈顶小于$A[i]$，那么$A[i]$应该是栈顶右子树的树根，将$A[i]$入栈，同时这时候，应该将$A[i]$的父亲记录下来，因为这时已经遍历到了右子树了，如果再遇到比它父亲小的数，那说明有矛盾，返回false。遍历完$A$之后如果没矛盾，则返回true。代码如下：

import java.util.ArrayDeque;
import java.util.Deque;

public class Solution {
    /**
     * @param preorder: List[int]
     * @return: return a boolean
     */
    public boolean verifyPreorder(int[] preorder) {
        // write your code here
        // 记录遍历到的子树的树根，也就是当前节点的下界
        int min = Integer.MIN_VALUE;
        
        Deque<Integer> stack = new ArrayDeque<>();
        for (int i = 0; i < preorder.length; i++) {
        	// 栈空则直接入栈
            if (stack.isEmpty()) {
                stack.push(preorder[i]);
                continue;
            }
            
            // 如果接下来的数小于了当前遍历的子树的树根，则返回false
            if (preorder[i] <= min) {
                return false;
            }
            
            // 遇到大于栈顶的数了，说明遍历到某个子树的右子树了，找到树根，并存储其值到min里
            if (stack.peek() > preorder[i]) {
                stack.push(preorder[i]);
            } else {
                while (!stack.isEmpty() && stack.peek() < preorder[i]) {
                    min = stack.pop();
                }
                
                stack.push(preorder[i]);
            }
        }
        
        return true;
    }
}

时间复杂度$O(n)$，空间$O(h)$，$h$是树的高度。