nowcoder106I Neat Tree (单调栈)

 1 #include<bits/stdc++.h>
 2 #define pa pair<int,int>
 3 #define CLR(a,x) memset(a,x,sizeof(a))
 4 using namespace std;
 5 typedef long long ll;
 6 const int maxn=1e6+10;
 7 
 8 inline ll rd(){
 9     ll x=0;char c=getchar();int neg=1;
10     while(c<‘0‘||c>‘9‘){if(c==‘-‘) neg=-1;c=getchar();}
11     while(c>=‘0‘&&c<=‘9‘) x=x*10+c-‘0‘,c=getchar();
12     return x*neg;
13 }
14 
15 int N,a[maxn];
16 int stk[maxn][2],sh,mm[maxn];
17 ll ans;
18 
19 int main(){
20     //freopen(".in","r",stdin);
21     int i,j,k;
22     while(~scanf("%d",&N)){
23         for(i=1;i<=N;i++) a[i]=rd();
24         CLR(mm,0);ans=0;
25         for(i=1;i<=N;i++){
26             while(sh&&stk[sh][0]<a[i]){
27                 mm[stk[sh--][1]]=i-1;
28             }
29             stk[++sh][0]=a[i],stk[sh][1]=i;
30         }for(;sh;--sh) mm[stk[sh][1]]=N;
31         
32         for(i=N;i;i--){
33             while(sh&&stk[sh][0]<=a[i]){
34                 if(mm[stk[sh][1]]) ans+=1ll*(stk[sh][1]-i)*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
35                 sh--;
36             }
37             stk[++sh][0]=a[i],stk[sh][1]=i;
38         }for(;sh;--sh) if(mm[stk[sh][1]]) ans+=1ll*stk[sh][1]*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
39         
40         CLR(mm,0);
41         for(i=1;i<=N;i++){
42             while(sh&&stk[sh][0]>a[i]){
43                 mm[stk[sh--][1]]=i-1;
44             }
45             stk[++sh][0]=a[i],stk[sh][1]=i;
46         }for(;sh;--sh) mm[stk[sh][1]]=N;
47         
48         for(i=N;i;i--){
49             while(sh&&stk[sh][0]>=a[i]){
50                 if(mm[stk[sh][1]]) ans-=1ll*(stk[sh][1]-i)*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
51                 sh--;
52             }
53             stk[++sh][0]=a[i],stk[sh][1]=i;
54         }for(;sh;--sh) if(mm[stk[sh][1]]) ans-=1ll*stk[sh][1]*(mm[stk[sh][1]]-stk[sh][1]+1)*stk[sh][0];
55             
56         printf("%lld\n",ans);
57     }
58     
59     return 0;
60 }