题目链接:B - Minimum Sum (atcoder.jp)
题意:给你n个数,求出任意一个区间的数的最小值的加和。
这道题目显然暴力是不行的,我们不能选定一个区间然后求这个区间的数中最小值,我们就逆向思考,我们可以考虑每一个值做最小值的区间情况,举个例子来说:
对于5个数 3 4 5 4 3 来说,我们考虑满足第二个数4做最小值的区间的左右边界应该满足什么条件,容易发现我们需要找到他左边第一个比他小的数来作为满足题意区间的左边界,也就是第一个位置,以及右边第一个比他小的数来作为满足题意区间的右边界,也就是第5个数,则满足题意的左边界应该在2~i之间,满足条件的右边界应该在i~4之间,这样我们就找到了【2,2】,【2,3】,【2,4】三个区间,这样真的就可以了吗?其实并不是这样,我们再考虑第二个4,我们可以找到【2,4】,【3,4】,【4,4】三个区间,发现了什么问题没有,就是出现了重复现象,【2,4】这个区间被计算了2次,如果同一个最小值出现在一段连续的满足题意的区间中多次那么将会有区间被多次计算,那我们怎么消除这个错误呢?对于左边界我们仍然找他左边第一个小于他的数的位置,而对于右边界而言我们可以找他右边第一个小于等于他的位置,这样就可以解决这个问题了,这样对于第一个4我们找到了左区间边界范围为2,右区间范围为2~3,而对于第二个4而言,左区间范围为2~4,右区间范围为4,这样我们就可以找到【2,2】,【2,3】,【2,4】,【3,4】,【4,4】这5个区间,从而也可以使问题得到正确解决,下面是代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
const int N=1e6+10;
int st[N],tt,l[N],r[N];
long long a[N];
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int i=1;i<=n;i++)
{
while(tt&&a[st[tt]]>=a[i]) tt--;
l[i]=st[tt];
st[++tt]=i;
}
tt=0;
for(int i=n;i>=1;i--)
{
while(tt&&a[st[tt]]>a[i]) tt--;//注意此处找的是这个数右边第一个小于等于他的数的位置
if(tt) r[i]=st[tt];
else r[i]=n+1;
st[++tt]=i;
}
long long ans=0;
for(int i=1;i<=n;i++)
ans+=a[i]*(i-l[i])*(r[i]-i);
cout<<ans;
return 0;
}