http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1215
题意:
思路:
计算出以第i个数为最大值的区间范围,l_max[i]为左端点,r_max[i]为右端点,计算最小值同理可得。
计算出了区间范围,就可以计算出每个数对于答案的贡献值。
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = +; int n;
int a[maxn]; int sta[maxn];
ll l_max[maxn],r_max[maxn];
ll l_min[maxn],r_min[maxn]; int main()
{
//freopen("in.txt","r",stdin);
while(~scanf("%d",&n))
{
for(int i=;i<=n;i++) scanf("%d",&a[i]); int top = ;
for(int i=;i<=n;i++)
{
while(top && a[sta[top]]<=a[i]) top--;
if(top==) l_max[i]=;
else l_max[i]=sta[top]+;
sta[++top]=i;
}
top=;
for(int i=n;i>=;i--)
{
while(top && a[sta[top]]<a[i]) top--; //这儿特别注意一下
if(top==) r_max[i]=n;
else r_max[i]=sta[top]-;
sta[++top]=i;
} top = ;
for(int i=;i<=n;i++)
{
while(top && a[sta[top]]>=a[i]) top--;
if(top==) l_min[i]=;
else l_min[i]=sta[top]+;
sta[++top]=i;
}
top=;
for(int i=n;i>=;i--)
{
while(top && a[sta[top]]>a[i]) top--;
if(top==) r_min[i]=n;
else r_min[i]=sta[top]-;
sta[++top]=i;
} ll ans =;
for(int i=;i<=n;i++) ans+=a[i]*((i-l_max[i]+)*(r_max[i]-i+)-(i-l_min[i]+)*(r_min[i]-i+));
printf("%lld\n",ans); }
return ;
}