0
点赞
收藏
分享

微信扫一扫

单调栈的理解

穆熙沐 2022-04-05 阅读 70

单调栈,顾名思义,一个单调递增或单调递减的栈,其关键在于满足入栈条件时的值

这道题,便需要使用到单调栈,思想便是:从左往右开始入栈,第一个肯定为-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]+" ");
		}
	}

}

单调栈的思想应当多理解。当需要找“某一数的某方向上的第一个比该数大(小)的数” 时,便可使用单调栈来解决。

 

举报

相关推荐

0 条评论