ACwing-830
例题:
给定一个长度为N的整数数列,输出每个数左边第一个比它小的数,如果不存在则输出-1。
输入格式
第一行包含整数N,表示数列长度。
第二行包含N个整数,表示整数数列。
输出格式
共一行,包含N个整数,其中第i个数表示第i个数的左边第一个比它小的数,如果不存在则输出-1。
数据范围
1≤N≤105
1≤数列中元素≤109
输入样例:
5
3 4 2 7 5
输出样例:
-1 3 -1 2 2
解题思路
用单调递增栈维护,当比栈顶大时,直接输出栈顶元素,并入栈;
比栈顶元素小时,则出栈,直到比栈顶元素小,重复上一步,或者为空输出-1,入栈;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Stack;
public class Main {
static BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
public static void main(String[] args) throws Exception {
int n = Integer.valueOf(bf.readLine());
String[] str = bf.readLine().split(" ");
int[] a = new int[n];
Stack<Integer> stack = new Stack<Integer>();
for(int i = 0; i < n; i++) a[i] = Integer.valueOf(str[i]);
for(int i = 0; i < n; i++) {
while(!stack.isEmpty() && stack.peek() >= a[i]) stack.pop();
if(stack.isEmpty())bw.write("-1 ");
else bw.write(Integer.valueOf(stack.peek()).toString() + " ");
stack.push(a[i]);
}
bw.flush();
}
}
LeetCode-42.接雨水
利用单调栈,每次与两侧的比,求出面积
class Solution {
public int trap(int[] height) {
int len = height.length;
if(len == 0 || len == 1)return 0;
Stack<Integer> stack = new Stack<>();
int ans = 0;
for(int i = 0; i < len; i++){
while(!stack.isEmpty() && height[stack.peek()] <= height[i]){
//将相等的去掉,因为相等的不用算面积
if(!stack.isEmpty() && height[stack.peek()] == height[i])stack.pop();
else{//算中间的面积
int temp = stack.pop();
if(!stack.isEmpty()){
ans += (Math.min(height[stack.peek()],height[i])-height[temp]) * (i - stack.peek() - 1);
}
}
}
stack.push(i);
}
return ans;
}
}