单调栈
就是利用一个栈来维持数据的单调性,任意时刻,栈中存放的元素都会是单调的。
单调栈适用于去解决在一串数字中,要获得某个数字其左右两边大小信息的问题,通常是可以将O(n^2)的问题去化为O(n)来解决。
比如,要获得所有数字左边离它最近的小于它的数,和右边离它最近的小于它的数这种问题,就可以使用单调栈。
让栈内保持单调递增,如果遇到小于栈顶元素的数(也就是它加入后会破坏单调性的情况),就出栈,然后记录栈顶元素的信息。对于栈顶元素来说,在栈中它下面压着的数是左边离它最近的小于它的数,而当前让它出栈的元素是右边离它最近的小于它的数。
代码如下:
public class Monotone_stack_code {
public static int[][] process(int[] nums) {
int[][] arr = new int[nums.length][2];
Stack<Integer> stack = new Stack<>(); //栈中存放的是nums的坐标
for (int i = 0; i < nums.length; i++) {
if (stack.isEmpty()) {
stack.push(i);
} else {
while (!stack.isEmpty()) {
//遇到可以破坏栈中的单调递增的数
if (nums[i] < nums[stack.peek()]) {
int cur = stack.pop();
//在该元素之前没有元素了,那就说明前面没有比它小的,记为-1
if (stack.isEmpty()) {
arr[cur][0] = -1;
} else {
arr[cur][0] = nums[stack.peek()];
}
//让它出栈的元素为它右边离它最近小于它的元素
arr[cur][1] = nums[i];
} else {
break;
}
}
stack.push(i);
}
}
//对于单调栈中剩余的元素进行处理,此时栈中是一个单调递增序列
while (!stack.isEmpty()) {
int cur = stack.pop();
//在该元素之前没有元素了,那就说明前面没有比它小的,记为-1
if (stack.isEmpty()) {
arr[cur][0] = -1;
} else {
arr[cur][0] = nums[stack.peek()];
}
//右边没有比它们小的了,都置为-1
arr[cur][1] = -1;
}
return arr;
}
public static void main(String[] args) {
int[] nums = {6, 7, 9, 8, 3, 5};
int[][] arr = process(nums);
System.out.println(Arrays.toString(nums));
for (int i = 0; i < arr.length; i++) {
System.out.println("" + nums[i] + " : " + Arrays.toString(arr[i]));
}
}
}

当然,这里是对于没有重复数字且都为正数的数而言。如果是存在负数的话,就不置为-1即可,对于重复数的话,stack中存放一个List,将每个等于该值的数的坐标放入List即可。
3196

被折叠的 条评论
为什么被折叠?



