题意:
给出一个1-N的排列,求
∑
l
=
1
N
∑
r
=
l
N
m
i
n
(
a
l
,
a
l
+
1
,
.
.
.
,
a
r
)
\displaystyle \sum^{N}_{l=1}\sum^{N}_{r=l}min(a_l,a_{l+1},...,a_r)
l=1∑Nr=l∑Nmin(al,al+1,...,ar)。
题解:
直接暴力很好想,但是会T。
这时候需要用到一种叫做单调栈的东西。
我们可以这样思考:求出每个位置作为最小值时可以到达的最左端和最右端。
那么可以用栈来维护一个单调递增的序列,每次访问一个元素时如果不满足单调递增,即栈顶大于等于该元素时就出栈。最后栈要么为空,要么栈顶小于当前元素,然后栈顶右边到当前元素一定大于等于当前值。
然后求答案时就可以直接左边*右边。
Upd:代码作了修改。其实不能同时取等的,否则会重复。
否则遇到
5
4 4 4 4 4
这种数据就挂了。
代码:
#include<cstdio>
#define maxn 200005
#define LL long long
int n,a[maxn],l[maxn],r[maxn],st[maxn],tp;
LL ans;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
{
while(tp&&a[st[tp]]>=a[i]) tp--;
l[i]=st[tp]; st[++tp]=i;
}
tp=0; st[0]=n+1;
for(int i=n;i;i--)
{
while(tp&&a[st[tp]]>a[i]) tp--;
r[i]=st[tp]; st[++tp]=i;
}
for(int i=1;i<=n;i++) ans+=1ll*a[i]*(i-l[i])*(r[i]-i);
printf("%lld\n",ans);
}