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 values of all subsegments of this array.
For example, the imbalance value of array [1, 4, 1] is 9, because there are 6 different 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.
The first line contains one integer n (1 ≤ n ≤ 106) — size of the array a.
The second line contains n integers a1, a2... an (1 ≤ ai ≤ 106) — elements of the array.
Print one integer — the imbalance value of a.
3 1 4 1
9
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
using namespace std;
#define ll __int64
int a[1000005];
int l[1000005];
int r[1000005];
int main()
{
int n;
while(~scanf("%d",&n))
{
ll output=0;
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<=n;i++)l[i]=r[i]=i;
for(int i=2;i<=n;i++)
{
int now=i;
while(now>1&&a[i]>=a[now-1])now=l[now-1];
l[i]=now;
}
for(int i=n-1;i>=1;i--)
{
int now=i;
while(now<n&&a[i]>a[now+1])now=r[now+1];
r[i]=now;
}
for(int i=1;i<=n;i++)output+=(ll)a[i]*(ll)(i-l[i]+1)*(ll)(r[i]-i+1);
for(int i=1;i<=n;i++)l[i]=r[i]=i;
for(int i=2;i<=n;i++)
{
int now=i;
while(now>1&&a[i]<=a[now-1])now=l[now-1];
l[i]=now;
}
for(int i=n-1;i>=1;i--)
{
int now=i;
while(now<n&&a[i]<a[now+1])now=r[now+1];
r[i]=now;
}
for(int i=1;i<=n;i++)output-=(ll)a[i]*(ll)(i-l[i]+1)*(ll)(r[i]-i+1);
printf("%I64d\n",output);
}
}