您的位置:首页 > 其它

51nod 1215 数组的宽度(单调栈)

2017-10-03 20:57 393 查看
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 = 50000+5;

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=1;i<=n;i++)  scanf("%d",&a[i]);

int top = 0;
for(int i=1;i<=n;i++)
{
while(top && a[sta[top]]<=a[i])  top--;
if(top==0)  l_max[i]=1;
else l_max[i]=sta[top]+1;
sta[++top]=i;
}
top=0;
for(int i=n;i>=1;i--)
{
while(top && a[sta[top]]<a[i])  top--;  //这儿特别注意一下
if(top==0)  r_max[i]=n;
else r_max[i]=sta[top]-1;
sta[++top]=i;
}

top = 0;
for(int i=1;i<=n;i++)
{
while(top && a[sta[top]]>=a[i])  top--;
if(top==0)  l_min[i]=1;
else l_min[i]=sta[top]+1;
sta[++top]=i;
}
top=0;
for(int i=n;i>=1;i--)
{
while(top && a[sta[top]]>a[i])  top--;
if(top==0)  r_min[i]=n;
else r_min[i]=sta[top]-1;
sta[++top]=i;
}

ll ans =0;
for(int i=1;i<=n;i++)  ans+=a[i]*((i-l_max[i]+1)*(r_max[i]-i+1)-(i-l_min[i]+1)*(r_min[i]-i+1));
printf("%lld\n",ans);

}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: