【单调栈】三校联考 10.15 —— array

题目描述

在放完棋子之后，dirty 又开始了新的游戏。
现在他拥有一个长为 n 的数组 A，他定义第 i 个位置的分值为 i − k + 1，其中 k 需要满足：
对于任意满足 k ≤ j ≤ i 的 j，有 A[k] ≤ A[j] ≤ A[i]。当对于第 i 个数，有多个 k 满足条件时，取
能获得较大分值的 k。
现在，dirty 想要知道 A 数组中分值最大的位置对应的分值为多少。

分析

首先思考暴力算法,从当前点i,向右边枚举至最远能满足要求即可.注意可以稍微优化以下,提前退出枚举.

然后正解就可以利用单调栈来进行维护信息,代替向右的枚举.
从右往左扫,单调栈维护单调递减的元素,并且储存在栈中元素到上一个栈中元素的区间中,最靠左的能够作为区间最短点的位置,以及这个位置的值.在弹出栈顶的时候,就可以计算出这个元素的答案.注意在此过程中维护上述的两个信息.

#include<cstdio>
#include<algorithm>
using namespace std;

#define lch rt<<1
#define rch rt<<1|1
const int MAXN=1e7+5;
const int INF=0x3f3f3f3f;

int n,ans=1;
int a[MAXN],pos[MAXN],mn[MAXN];
int sta[MAXN],top;

int main(){
    freopen("array.in","r",stdin);
    freopen("array.out","w",stdout);

    scanf("%d",&n); n++;
    a[1]=INF;
    for(int i=2;i<=n;i++) scanf("%d",&a[i]);
    sta[++top]=n,pos[top]=n,mn[top]=a[n];
    for(int i=n-1;i>0;i--){
        int tpos=i+1,tmin=a[i+1];
        while(top&&a[i]>a[sta[top]]){
            ans=max(ans,sta[top]-tpos+1);
            if(tmin>mn[top]) tmin=mn[top],tpos=pos[top];
            top--;
        }
        if(!top){
			sta[++top]=i,pos[top]=i,mn[top]=a[i];
			continue ;
		}
		if(a[i]<tmin){
			tmin=a[i];
			tpos=i;
		}
		sta[++top]=i,pos[top]=tpos,mn[top]=tmin;
    }
    printf("%d",ans);

    fclose(stdin);
    fclose(stdout);
}