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+;

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

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

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

         for(i=N;i;i--){
             while(sh&&stk[sh][]<=a[i]){
                 if(mm[stk[sh][]]) ans+=1ll*(stk[sh][]-i)*(mm[stk[sh][]]-stk[sh][]+)*stk[sh][];
                 sh--;
             }
             stk[++sh][]=a[i],stk[sh][]=i;
         }for(;sh;--sh) if(mm[stk[sh][]]) ans+=1ll*stk[sh][]*(mm[stk[sh][]]-stk[sh][]+)*stk[sh][];

         CLR(mm,);
         for(i=;i<=N;i++){
             while(sh&&stk[sh][]>a[i]){
                 mm[stk[sh--][]]=i-;
             }
             stk[++sh][]=a[i],stk[sh][]=i;
         }for(;sh;--sh) mm[stk[sh][]]=N;

         for(i=N;i;i--){
             while(sh&&stk[sh][]>=a[i]){
                 if(mm[stk[sh][]]) ans-=1ll*(stk[sh][]-i)*(mm[stk[sh][]]-stk[sh][]+)*stk[sh][];
                 sh--;
             }
             stk[++sh][]=a[i],stk[sh][]=i;
         }for(;sh;--sh) if(mm[stk[sh][]]) ans-=1ll*stk[sh][]*(mm[stk[sh][]]-stk[sh][]+)*stk[sh][];

         printf("%lld\n",ans);
     }

     return ;
 }