nowcoder106I Neat Tree (单调栈)

Richard神犇出给nowcoder的题

用单调栈找到每个点它向右和向左的第一个大于或小于它的位置，然后它作为最大值/最小值的区间就要在这个范围里，那么它的贡献就是这个区间长度乘一乘再减一减

注意一下值相等的时候怎么处理

#include<bits/stdc++.h>
#define pa pair<int,int>
#define CLR(a,x) memset(a,x,sizeof(a))
using namespace std;
typedef long long ll;
const int maxn=1e6+10;

inline ll rd(){
    ll x=0;char c=getchar();int neg=1;
    while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();}
    while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*neg;
}

int N,a[maxn];
int stk[maxn][2],sh,mm[maxn];
ll ans;

int main(){
    //freopen(".in","r",stdin);
    int i,j,k;
    while(~scanf("%d",&N)){
        for(i=1;i<=N;i++) a[i]=rd();
        CLR(mm,0);ans=0;
        for(i=1;i<=N;i++){
            while(sh&&stk[sh][0]<a[i]){
                mm[stk[sh--][1]]=i-1;
            }
            stk[++sh][0]=a[i],stk[sh][1]=i;
        }for(;sh;--sh) mm[stk[sh][1]]=N;
        
        for(i=N;i;i--){
            while(sh&&stk[sh][0]<=a[i]){
                if(mm[stk[sh][1]]) ans+=1ll*(stk[sh][1]-i)*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
                sh--;
            }
            stk[++sh][0]=a[i],stk[sh][1]=i;
        }for(;sh;--sh) if(mm[stk[sh][1]]) ans+=1ll*stk[sh][1]*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
        
        CLR(mm,0);
        for(i=1;i<=N;i++){
            while(sh&&stk[sh][0]>a[i]){
                mm[stk[sh--][1]]=i-1;
            }
            stk[++sh][0]=a[i],stk[sh][1]=i;
        }for(;sh;--sh) mm[stk[sh][1]]=N;
        
        for(i=N;i;i--){
            while(sh&&stk[sh][0]>=a[i]){
                if(mm[stk[sh][1]]) ans-=1ll*(stk[sh][1]-i)*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
                sh--;
            }
            stk[++sh][0]=a[i],stk[sh][1]=i;
        }for(;sh;--sh) if(mm[stk[sh][1]]) ans-=1ll*stk[sh][1]*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
            
        printf("%lld
",ans);
    }
    
    return 0;
}