题目地址:
https://leetcode.com/problems/verify-preorder-sequence-in-binary-search-tree/
给定一个长 n n n的数组 A A A,判断其是否可能是一个二叉搜索树的前序遍历。题目保证 A A A里没有重复数字。
栈的方法参考https://blog.csdn.net/qq_46105170/article/details/109635042。下面给出一个递归写法:
可以模仿由先序遍历还原二叉搜索树的方式来做。用一个全局变量记录遍历 A A A已经到达的位置,用两个变量记录节点取值的范围,然后模拟建树的过程。如果在范围之内,则new出节点,然后递归建左右子树。最后如果 A A A中有数字没用上,则说明不合法,返回false。代码如下:
public class Solution {
private int idx;
public boolean verifyPreorder(int[] preorder) {
dfs(preorder, Integer.MIN_VALUE, Integer.MAX_VALUE);
return idx == preorder.length;
}
private void dfs(int[] preorder, int min, int max) {
if (idx == preorder.length) {
return;
}
int val = preorder[idx];
if (min < preorder[idx] && preorder[idx] < max) {
// 在范围内,则new子节点,将idx向后挪一位
idx++;
// 其左子树的范围应该是min到当前节点;右子树范围应该是当前节点到max
dfs(preorder, min, val);
dfs(preorder, val, max);
}
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( h ) O(h) O(h), h h h为树的高度,也是递归栈深度。