51nod 1215 数组的宽度（单调栈）

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
",ans);

    }
    return 0;
}