51nod 1215：数组的宽度 单调栈

1215 数组的宽度

题目来源：  Javaman

基准时间限制：1 秒 空间限制：131072 KB 分值: 80  难度：5级算法题

 收藏

 关注

N个整数组成的数组，定义子数组a[i]..a[j]的宽度为：max(a[i]..a[j]) - min(a[i]..a[j])，求所有子数组的宽度和。

Input

第1行：1个数N，表示数组的长度。(1 <= N <= 50000)
第2 - N + 1行：每行1个数，表示数组中的元素(1 <= A[i] <= 50000)

Output

输出所有子数组的宽度和。

Input示例

5
1
2
3
4
5

Output示例

20

现在发现了，如果我缘分地做出了某一道题，我一定是以最最麻烦的方式做出来的。

这道题可以用单调栈来求出每一个元素能够控制的区间，即在哪些区间内这个元素是最大值，哪些区间内这个元素是最小值，然后对答案计算贡献。

然后我就跑了四遍。。。其实两遍完全可以跑出来的，一前一后。。。

然后计算区间的时候，左边弹栈了的话，右边就不能弹栈。保证两轮总有一个控制，不能全控制，也不能都不控制。

代码：

#pragma warning(disable:4996)
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <string>
#include <cstdio>
#include <cmath>
#include <queue>
#include <stack>
#include <deque>
#include <set>
#include <map>
using namespace std;

#define INF 0x3fffffff
typedef long long ll;

const ll mod = 1e9 + 7;
const ll maxn = 50010;

int n;
int A[maxn], num[maxn], sta[maxn], rimax[maxn], rimin[maxn], lemax[maxn], lemin[maxn];

void input()
{
	int i;
	scanf("%d", &n);

	for (i = 1; i <= n; i++)
	{
		scanf("%d", &A[i]);
		num[A[i]]++;
	}
}

void solve()
{
	int i;
	int top = 0;
	
	memset(sta, 0, sizeof(sta));
	A[n + 1] = -1;
	for (i = 1; i <= n + 1; i++)
	{
		if (top == 0 || A[i] > A[sta[top]])
		{
			sta[++top] = i;
			continue;
		}
		while (top >= 1 && A[i] < A[sta[top]])
		{
			rimin[sta[top]] = i - 1;		
			top--;
		}
		top++;
		sta[top] = i;
	}

	memset(sta, 0, sizeof(sta));
	A[0] = -1;
	for (i = n; i >= 0; i--)
	{
		if (top == 0 || A[i] > A[sta[top]])
		{
			sta[++top] = i;
			continue;
		}
		while (top >= 1 && A[i] <= A[sta[top]])
		{
			lemin[sta[top]] = i + 1;		
			top--;
		}
		top++;
		sta[top] = i;
	}
	memset(sta, 0, sizeof(sta));
	A[0] = 50005;
	for (i = n; i >= 0; i--)
	{
		if (top == 0 || A[i] < A[sta[top]])
		{
			sta[++top] = i;
			continue;
		}
		while (top >= 1 && A[i] >= A[sta[top]])
		{
			lemax[sta[top]] = i + 1;		
			top--;
		}
		top++;
		sta[top] = i;
	}

	memset(sta, 0, sizeof(sta));
	A[n + 1] = 50005;
	for (i = 1; i <= n + 1; i++)
	{
		if (top == 0 || A[i] < A[sta[top]])
		{
			sta[++top] = i;
			continue;
		}

		while (top >= 1 && A[i] > A[sta[top]])
		{
			rimax[sta[top]] = i - 1;
			top--;
		}
		top++;
		sta[top] = i;
	}

	ll ans = 0;
	for (i = 1; i <= n; i++)
	{
		ll d1 = (rimax[i] - i + 1)*(i - lemax[i] + 1);
		ll d2 = (rimin[i] - i + 1)*(i - lemin[i] + 1);
		ans += (d1) * (ll)A[i];
		ans -= (d2) * (ll)A[i];
	}

	printf("%lld", ans);
}

int main()
{
	//freopen("i.txt","r",stdin);
	//freopen("o.txt","w",stdout);

	input();
	solve();

	//system("pause");
	return 0;
}