单调栈的使用:
求解数组中元素右边第一个比它小的元素的下标,从前往后,构造单调递增栈;
求解数组中元素右边第一个比它大的元素的下标,从前往后,构造单调递减栈;
求解数组中元素左边第一个比它小的元素的下标,从后往前,构造单调递减栈;
求解数组中元素左边第一个比它小的元素的下标,从后往前,构造单调递增栈。
场景:
给定一个整数数组 A
,坡是元组 (i, j)
,其中 i < j
且 A[i] <= A[j]
。这样的坡的宽度为 j - i
。
找出 A
中的坡的最大宽度,如果不存在,返回 0 。
示例 1:
输入:[6,0,8,2,1,5] 输出:4 解释: 最大宽度的坡为 (i, j) = (1, 5): A[1] = 0 且 A[5] = 5.
示例 2:
输入:[9,8,1,0,1,9,4,0,4,1] 输出:7 解释: 最大宽度的坡为 (i, j) = (2, 9): A[2] = 1 且 A[9] = 1.
提示:
2 <= A.length <= 50000
0 <= A[i] <= 50000
思路:对于数组中的某一个数
题目是让找左侧数,第一个比他小的数。换种思路想,从普遍题的角度去思考,但凡找数组的某一个数的左侧、右侧第一个比他大或者小的,都可以试着用单调栈来解决问题,维护一个单调递增或者单调递减的栈,然后试着从左或者从右去遍历原数组,求解相应问题。
如此题中,求最大的宽度,从后朝前遍历比较合理,那么相应的构造的栈应该是从左往右的,那么是构造递增还是递减的呢?比如构造的是递增的话1,2,3,... 当前考虑 9 , 那么去遍历构造好的栈,9>3,那么之后的2,1就没有意义了,所以这里应该构造单调递减的栈。3,2,1... 当前考虑9,9>1满足,那么更新当前宽度最大值,之后比较9和2...直到栈空,或者遍历完毕。
public int maxWidthRamp(int[] A) {
Stack<Integer> stack = new Stack<Integer>();
// 递减栈,如果存值的话,那么一起还要存对应的下标
// 如果存的是下标的话,可以通过A[x]找到值,所以这里采取存下标的方式,可以节约一部分空间
int len = A.length;
if(len<2)return 0; // 特判处理
// 预处理
stack.add(0); // 下标0
for(int i = 1 ; i < len ; i++){
// 单调递减
if(A[i]<=A[stack.peek()]){
stack.add(i);
}
}
int res = Integer.MIN_VALUE;
// 预处理完毕,从后朝前遍历。
for(int i = len-1 ; i >= 0 ; i--){
while(!stack.isEmpty() && A[i]>=A[stack.peek()]){
res = Math.max(res, i-stack.pop());
}
}
return res;
}