Imbalanced Array CodeForces - 817D （思维+单调栈）

Posted 2021-02-06 qieqiemin

You are given an array a consisting of n elements. The imbalance value of some subsegment of this array is the difference between the maximum and minimum element from this segment. The imbalance value of the array is the sum of imbalance valuesof all subsegments of this array.

For example, the imbalance value of array [1, 4, 1] is 9, because there are 6different subsegments of this array:

• [1] (from index 1 to index 1), imbalance value is 0;
• [1, 4] (from index 1 to index 2), imbalance value is 3;
• [1, 4, 1] (from index 1 to index 3), imbalance value is 3;
• [4] (from index 2 to index 2), imbalance value is 0;
• [4, 1] (from index 2 to index 3), imbalance value is 3;
• [1] (from index 3 to index 3), imbalance value is 0;

You have to determine the imbalance value of the array a.

Input

The first line contains one integer n (1 ≤ n ≤ 10⁶) — size of the array a.

The second line contains n integers a₁, a₂... a_n (1 ≤ a_i ≤ 10⁶) — elements of the array.

Output

Print one integer — the imbalance value of a.

Example

Input

3
1 4 1

Output

9

题意：
给定一个含有N个数的数组，求这个数组的所有连续区间的不平衡值的总和。
一个区间的不平衡值为这个区间中的最大值减去最小值。
思路：
我们可以这样思考，这个题目的答案可以这样得来，
对于每一个a[i]，它对答案的贡献值为以a[i]为区间最大值的区间数量*a[i]-a[i]*以a[i]为区间最小值的区间数量。
ans=sum{ contribution（a[i]）}

那么我们的难题为如何计算以a[i]为区间最大/最小值的区间数量。
这种模型我们很容易想到利用单调栈来O(N)求出。
我们定义两个数组,l[n+5],r[n+5]，l[i]和r[i] 分别维护的是从a[i]元素开始向左和向右第一个比a[i]小的元素的位置。（注意边界）
然后我们可以通过l[i]和r[i]来求出区间数量，然后我们再利用单调栈求出从a[i]元素开始向左和向右第一个比a[i]大的元素的位置

然后我们愉快的计算每一个的贡献然后加起来就是我们要求的答案。
细节见代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#define rt return
#define sz(a) int(a.size())
#define all(a) a.begin(), a.end()
#define rep(i,x,n) for(int i=x;i<n;i++)
#define repd(i,x,n) for(int i=x;i<=n;i++)
#define pii pair<int,int>
#define pll pair<long long ,long long>
#define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define MS0(X) memset((X), 0, sizeof((X)))
#define MSC0(X) memset((X), ‘‘, sizeof((X)))
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define eps 1e-6
#define gg(x) getInt(&x)
#define db(x) cout<<"== [ "<<x<<" ] =="<<endl;
using namespace std;
typedef long long ll;
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
ll powmod(ll a,ll b,ll MOD){ll ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
inline void getInt(int* p);
const int maxn=1000010;
const int inf=0x3f3f3f3f;
/*** TEMPLATE CODE * * STARTS HERE ***/
int n;
int a[maxn];
int l[maxn];
int r[maxn];
ll ans=0ll;
int main()
{
    gg(n);
    repd(i,1,n)
    {
        gg(a[i]);
    }
    stack<int> s;
    repd(i,1,n)
    {
        while(s.size()&&a[s.top()]>a[i])
        {
            s.pop();
        }
        if(s.size())
        {
            l[i]=s.top();
        }else
        {
            l[i]=0;
        }
        s.push(i);
    }
    while(s.size())
    {
        s.pop();
    }
    for(int i=n;i>=1;i--)
    {
        while(s.size()&&a[s.top()]>=a[i])
        {
            s.pop();
        }
        if(s.size())
        {
            r[i]=s.top();
        }else
        {
            r[i]=n+1;
        }
        s.push(i);
    }
    repd(i,1,n)
    {
        ans=ans-1ll*a[i]*(i-l[i])*(r[i]-i);
    }
//    db(ans);
    while(s.size())
    {
        s.pop();
    }
    repd(i,1,n)
    {
        while(s.size()&&a[s.top()]<a[i])
        {
            s.pop();
        }
        if(s.size())
        {
            l[i]=s.top();
        }else
        {
            l[i]=0;
        }
        s.push(i);
    }
    while(s.size())
    {
        s.pop();
    }
    for(int i=n;i>=1;i--)
    {
        while(s.size()&&a[s.top()]<=a[i])
        {
            s.pop();
        }
        if(s.size())
        {
            r[i]=s.top();
        }else
        {
            r[i]=n+1;
        }
        s.push(i);
    }

    repd(i,1,n)
    {
        ans=ans+1ll*a[i]*(i-l[i])*(r[i]-i);
    }
    printf("%lld
",ans );
    return 0;
}

inline void getInt(int* p) {
    char ch;
    do {
        ch = getchar();
    } while (ch == ‘ ‘ || ch == ‘
‘);
    if (ch == ‘-‘) {
        *p = -(getchar() - ‘0‘);
        while ((ch = getchar()) >= ‘0‘ && ch <= ‘9‘) {
            *p = *p * 10 - ch + ‘0‘;
        }
    }
    else {
        *p = ch - ‘0‘;
        while ((ch = getchar()) >= ‘0‘ && ch <= ‘9‘) {
            *p = *p * 10 + ch - ‘0‘;
        }
    }
}