在数组中想找到一个数, 左边和右边比这个数小、 且离这个数最近的位置。
如果对每一个数都想求这样的信息, 能不能整体代价达到O(N)? 需要使用到单调栈结构。
当有重复值的时候,我们将重复的值的index,放到一起,弹出的时候一起弹出。
整体逻辑:用一个栈辅助,栈里存着索引,对应的值是单调的,从栈顶往下是递减的(两边找最近的比他小的数)。如果说stack.top<arr[i],弹出此时的栈顶index = stack.pop()(记录下来作为某点index),然后它的右边最近比它小的数,就是arr[i],左边最近的比它小得数,就是弹出它自身后,前面一个(也就是此时的栈顶,如果此时栈为空,说明左面没有比它小的数,返回-1)
和窗口内最大值更新结构【重要】一样,都是先用while保证了大——小的结构,都是想i位置的情况,之后在add。除此之外,多了一个扫尾的工作。
package class04;
import java.util.List;
import java.util.ArrayList;
import java.util.Stack;
public class Code02_MonotonousStack {
//res res[i][0]表示左边离i位置最近的,比i小的数,的位置
//res res[i][1]表示右边。。。。。。。。。。。。
//arr内无重复的情况
public static int[][] getNearLessNoRepeat(int[] arr) {
int[][] res = new int[arr.length][2];
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < arr.length; i++) {//遍历到i位置的arr[i]怎么处理
while (!stack.isEmpty() && arr[stack.peek()] > arr[i]) {
int popIndex = stack.pop();
int leftLessIndex = stack.isEmpty() ? -1 : stack.peek();
res[popIndex][0] = leftLessIndex;
res[popIndex][1] = i;
}
stack.push(i);
}
//扫尾阶段
while (!stack.isEmpty()) {
int popIndex = stack.pop();
int leftLessIndex = stack.isEmpty() ? -1 : stack.peek();
res[popIndex][0] = leftLessIndex;
res[popIndex][1] = -1;//右边必然是0
}
return res;
}
//arr内有重复的情况
public static int[][] getNearLess(int[] arr) {
int[][] res = new int[arr.length][2];
Stack<List<Integer>> stack = new Stack<>();//栈里面放的是列表,因为如果重复了,要放到一起。
for (int i = 0; i < arr.length; i++) {
while (!stack.isEmpty() && arr[stack.peek().get(0)] > arr[i]) {
List<Integer> popIs = stack.pop();//注意,弹出的是一个列表,要遍历这个列表中的所有index
// 取位于下面位置的列表中,最晚加入的那个
int leftLessIndex = stack.isEmpty() ? -1 : stack.peek().get(
stack.peek().size() - 1);
for (Integer popi : popIs) {
res[popi][0] = leftLessIndex;
res[popi][1] = i;
}
}
//i位置进栈,此时i进栈肯定能满足整个栈 从下到上 依次为从小到大
if (!stack.isEmpty() && arr[stack.peek().get(0)] == arr[i]) {
stack.peek().add(Integer.valueOf(i));//如果相等,加到peek的list中
} else {//如果不等,新建一个列表,存入这个数,压到栈中
ArrayList<Integer> list = new ArrayList<>();
list.add(i);
stack.push(list);
}
}
// 扫尾
while (!stack.isEmpty()) {
List<Integer> popIs = stack.pop();
// 取位于下面位置的列表中,最晚加入的那个
int leftLessIndex = stack.isEmpty() ? -1 : stack.peek().get(
stack.peek().size() - 1);
for (Integer popi : popIs) {
res[popi][0] = leftLessIndex;
res[popi][1] = -1;
}
}
return res;
}
// for test
public static int[] getRandomArrayNoRepeat(int size) {
int[] arr = new int[(int) (Math.random() * size) + 1];
for (int i = 0; i < arr.length; i++) {
arr[i] = i;
}
for (int i = 0; i < arr.length; i++) {
int swapIndex = (int) (Math.random() * arr.length);
int tmp = arr[swapIndex];
arr[swapIndex] = arr[i];
arr[i] = tmp;
}
return arr;
}
// for test
public static int[] getRandomArray(int size, int max) {
int[] arr = new int[(int) (Math.random() * size) + 1];
for (int i = 0; i < arr.length; i++) {
arr[i] = (int) (Math.random() * max) - (int) (Math.random() * max);
}
return arr;
}
// for test
public static int[][] rightWay(int[] arr) {
int[][] res = new int[arr.length][2];
for (int i = 0; i < arr.length; i++) {
int leftLessIndex = -1;
int rightLessIndex = -1;
int cur = i - 1;
while (cur >= 0) {
if (arr[cur] < arr[i]) {
leftLessIndex = cur;
break;
}
cur--;
}
cur = i + 1;
while (cur < arr.length) {
if (arr[cur] < arr[i]) {
rightLessIndex = cur;
break;
}
cur++;
}
res[i][0] = leftLessIndex;
res[i][1] = rightLessIndex;
}
return res;
}
// for test
public static boolean isEqual(int[][] res1, int[][] res2) {
if (res1.length != res2.length) {
return false;
}
for (int i = 0; i < res1.length; i++) {
if (res1[i][0] != res2[i][0] || res1[i][1] != res2[i][1]) {
return false;
}
}
return true;
}
// for test
public static void printArray(int[] arr) {
for (int i = 0; i < arr.length; i++) {
System.out.print(arr[i] + " ");
}
System.out.println();
}
public static void main(String[] args) {
int size = 10;
int max = 20;
int testTimes = 2000000;
for (int i = 0; i < testTimes; i++) {
int[] arr1 = getRandomArrayNoRepeat(size);
int[] arr2 = getRandomArray(size, max);
if (!isEqual(getNearLessNoRepeat(arr1), rightWay(arr1))) {
System.out.println("Oops!");
printArray(arr1);
break;
}
if (!isEqual(getNearLess(arr2), rightWay(arr2))) {
System.out.println("Oops!");
printArray(arr2);
break;
}
}
}
}
应用
定义: 数组中所有数都是正数,数组中累积和与最小值的乘积, 假设叫做指标A。
给定一个数组, 请返回子数组中, 指标A最大的值。
首先,需要求arr[i...j]的sum,把所有满足i位置为最小值的最长的数组,找出来。
最长的其实就是找到每个位置的左右离它最近的比它小的数,夹在中间就是这个题,所以用单调栈。
public static int max2(int[] arr) {
int size = arr.length;
int[] sums = new int[size];
sums[0] = arr[0];//先把sum求出来
for (int i = 1; i < size; i++) {
sums[i] = sums[i - 1] + arr[i];
}
int max = Integer.MIN_VALUE;
Stack<Integer> stack = new Stack<Integer>();
for (int i = 0; i < size; i++) {
while (!stack.isEmpty() && arr[stack.peek()] >= arr[i]) {
int j = stack.pop();
max = Math.max(max, (stack.isEmpty() ? sums[i - 1] : (sums[i - 1] - sums[stack.peek()])) * arr[j]);
}
stack.push(i);
}//扫尾工作
while (!stack.isEmpty()) {
int j = stack.pop();
max = Math.max(max, (stack.isEmpty() ? sums[size - 1] : (sums[size - 1] - sums[stack.peek()])) * arr[j]);
}
return max;
}