Educational Codeforces Round 23 D. Imbalanced Array(单调栈)

题目链接：Educational Codeforces Round 23 D. Imbalanced Array

题意：

给你n个数，定义一个区间的不平衡因子为该区间最大值-最小值。

然后问你这n个数所有的区间的不平衡因子和

题解：

对每一个数算贡献，a[i]的贡献为 当a[i]为最大值时的 a[i]*(i-l+1)*(r-i+1) - 当a[i]为最小值时的a[i]*(i-l+1)*(r-i+1)。

计算a[i]的l和r时，用单调栈维护。具体看代码，模拟一下就知道了。

然后把所有的贡献加起来。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
typedef long long ll;
typedef pair<int,int> P;

const int N=1e6+7;
int n,a[N],l[N],r[N],top;
ll ans;
P S[N];
void work()
{
    top=1,S[1]=P(N,0);
    F(i,1,n)
    {
        while(S[top].first<=a[i])top--;
        l[i]=S[top].second+1;
        S[++top]=P(a[i],i);
    }
    top=1,S[1]=P(N,n+1);
    for(int i=n;i;i--)
    {
        while(S[top].first<a[i])top--;
        r[i]=S[top].second-1;
        S[++top]=P(a[i],i);
    }
    F(i,1,n)ans+=1ll*a[i]*(i-l[i]+1)*(r[i]-i+1);
}

int main(){
    scanf("%d",&n);
    F(i,1,n)scanf("%d",a+i);
    work();
    F(i,1,n)a[i]*=-1;
    work();
    printf("%lld\n",ans);
    return 0;
}