标签：codeforce 相同 main array const 个数 序列 避免 mes
https://codeforces.com/contest/817/problem/D
单调栈
单调栈经典问题。
我们先求出以 a[ i ] 为最小值的左右最长拓展 L1[i] , R1[i]
那么以 a[ i ] 为最小值的区间个数就为 (L1[ i ] + 1) * (R1[ i ] + 1)
再求出以 a[ i ] 为最大值的左右最长拓展 L2[ i ] , R2[ i ]
那么以 a[ i ] 为最大值的区间个数就为 (L2[ i ] + 1) * (R2[ i ] + 1)
最后线性扫一遍即可
(为了避免相同值区间长度被重复计算，需采用左开右闭的维护方式)
#include<bits/stdc++.h> #define rep(i,a,n) for (int i=a;i<=n;i++) #define per(i,n,a) for (int i=n;i>=a;i--) #define int long long using namespace std; const int N = 1e6 + 10; int a[N] , sta[N] , l[N] , r[N] , ans; signed main() { int n ; cin >> n; rep(i , 1 , n) cin >> a[i]; int top = 0; rep(i , 1 , n) { while(top && a[sta[top]] >= a[i]) top -- ; if(!top) l[i] = i - 1; else l[i] = i - sta[top] - 1; sta[++ top] = i; } top = 0; per(i , n , 1) { while(top && a[sta[top]] > a[i]) top --; if(!top) r[i] = n - i; else r[i] = sta[top] - i - 1; sta[++ top] = i; } rep(i , 1 , n) ans -= a[i] * (l[i] + 1) * (r[i] + 1); top = 0; rep(i , 1 , n) { while(top && a[sta[top]] <= a[i]) top -- ; if(!top) l[i] = i - 1; else l[i] = i - sta[top] - 1; sta[++ top] = i; } top = 0; per(i , n , 1) { while(top && a[sta[top]] < a[i]) top --; if(!top) r[i] = n - i; else r[i] = sta[top] - i - 1; sta[++ top] = i; } rep(i , 1 , n) ans += a[i] * (l[i] + 1) * (r[i] + 1); cout << ans << ‘\n‘; return 0; }
标签：codeforce 相同 main array const 个数 序列 避免 mes
原文地址：https://www.cnblogs.com/StarRoadTang/p/13026457.html
