单调栈,顾名思义,一个单调递增或单调递减的栈,其关键在于满足入栈条件时的值
这道题,便需要使用到单调栈,思想便是:从左往右开始入栈,第一个肯定为-1(读题先),然后下一个数判断是否小于前一个数:是 则该数左边第一比该数大的数为栈顶元素,不是 则移除栈顶元素后继续判断。往后的数照旧。
import java.util.Scanner;
import java.util.Stack;
public class 百亿富翁 {
static long[] arr=new long[700005];
static long[] l=new long[700005]; //存放从左到右的值
static long[] r=new long[700005]; //存放从右到左的值
static Stack<Integer> s=new Stack<>();
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
for(int i=1;i<=n;i++) { //因为题目要求最后输出编号,所以从1开始,方便后面输出
arr[i]=sc.nextLong();
}
for(int i=1;i<=n;i++) {
//若栈不为空且该数大于栈顶的数则将栈顶的数删掉
while(s.size()!=0 && arr[s.peek()]<=arr[i]) {
s.pop();
}
if(!s.isEmpty()) {
l[i]=s.peek();
}else {
//如果为空,表示没有找到比他大的数,返回-1
l[i]=-1;
}
s.push(i); //将编号存入栈
}
while(!s.isEmpty()) {
s.pop();
}
for(int i=n;i>0;i--) {
while(s.size()!=0 && arr[s.peek()]<=arr[i]) {
s.pop();
}
if(!s.isEmpty()) {
r[i]=s.peek();
}else {
r[i]=-1;
}
s.push(i);
}
for(int i=1;i<=n;i++) {
System.out.print(l[i]+" ");
}
System.out.println();
for(int i=1;i<=n;i++) {
System.out.print(r[i]+" ");
}
}
}
单调栈的思想应当多理解。当需要找“某一数的某方向上的第一个比该数大(小)的数” 时,便可使用单调栈来解决。