单调栈
原型应用:求数组中每个数左边离它最近的比它大的和右边离她最近的比它大的数。
暴力解:每个位置两边遍历,时间复杂度O(n^2)。
单调栈:时间复杂度O(n)。
原型
所用的数据结构:栈(从栈底到栈顶从大到小),相同元素压在一起,压入栈的都是角标。
流程:
- 若arr[i]比栈顶元素小,压入栈;
- 若arr[i]与栈顶元素相同,压在一起(例如用链表);
- 若arr[i]比栈顶元素大,则将栈顶元素弹出,直到arr[i] <= 栈顶元素;
注意:在弹出的时候开始收集信息。弹出时,弹出元素右边比它大的是arr[i],左边比它大的是栈中它的下一个元素。
两个特殊情况:
- 当最后一个栈顶元素弹出时,其左边比它大的为null(没有);
- 当元素遍历完后,栈中若还有元素,则此时弹出的所有元素右边最大的值为null。
经典应用
解题大致思路:
- 将矩阵做变化,为之后将问题转化为数组求解做准备;
- 对于每一行数组,利用单调栈来找最大值。
对于每个数组的计算流程:
- 准备一个栈,它是从栈底到栈顶严格增大的(要找最小值,则是这种单调栈结构);
- 若arr[i]比栈顶元素大,直接压栈;
- 若arr[i] <= 栈顶元素,则弹出栈顶元素,直至arr[i]比栈顶元素大。在弹出栈时,需要计算包含该元素的面积。
贴代码:
public static int solution(int[][] arr) {
if (arr.length == 0 || arr[0].length == 0) return 0;
//先将矩阵处理
for (int i = 1; i < arr.length; i++) {
for (int j = 0; j < arr[0].length; j++) {
arr[i][j] = arr[i][j] == 0 ? 0 : arr[i - 1][j] + arr[i][j];
}
}
//按每一行的数组计算最大面积
int max = Integer.MIN_VALUE;
for (int i = 0; i < arr.length; i++) {
//计算数组的最大面积
Stack<Integer> stack = new Stack<>();
for (int j = 0; j < arr[0].length; j++) {
while (!stack.isEmpty() && arr[i][stack.peek()] >= arr[i][j]) {
int temp = stack.pop();
int k = stack.isEmpty() ? -1 : stack.peek();
max = Math.max(max, arr[i][temp] * (j - k - 1));
}
stack.push(j);
}
while (!stack.isEmpty()) {
int temp = stack.pop();
int k = stack.isEmpty() ? -1 : stack.peek();
max = Math.max(max, arr[i][temp] * (arr[0].length - k - 1));
}
}
return max;
}