【CF-817D】Imbalanced Array【单调栈】

题意

给你个a序列, 求a序列的任意连续子序列中最大值-最小值的值的和.

分析

我们考虑每一个数对答案的贡献, 在一段连续的区间中, 这个数要么作为最大值, 要么作为最小值, 这样我们就设$l[i]$是第 $i$ 个数它往左最大可以到的位置, $r[i]$是 $i$ 往右最大可以到的位置, 这样答案就加上 $ans+=a[i]\times （i-L[i]）\times （R[i]-i)$. 同理一样找最小的位置,$ans-=a[i]\times （i-L[i]）\times （R[i]-i)$

答案就是每个数作为最大值的和减去每个数作为最小值。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<stack>
using namespace std;

stack<int> st;

long long n,a[1000001],ans,l[1000001],r[1000001];

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	a[0]=1000000000;
	st.push(0);
	for(int i=1;i<=n;i++)
	{
		while(a[i]>=a[st.top()])
		{
			st.pop();
		}
		l[i]=st.top();
		st.push(i);
	}
	while(!st.empty()) st.pop();
	a[n+1]=1000000000;
	st.push(n+1);
	for(int i=n;i>=1;i--)
	{
		while(a[i]>a[st.top()])
		{
			st.pop();
		}
		r[i]=st.top();
		st.push(i);
	}
	for(int i=1;i<=n;i++)
	{
		ans+=a[i]*(r[i]-i)*(i-l[i]);
	}
	
	while(!st.empty()) st.pop();
	a[0]=0;
	st.push(0);
	for(int i=1;i<=n;i++)
	{
		while(a[i]<=a[st.top()])
		{
			st.pop();
		}
		l[i]=st.top();
		st.push(i);
	}
	while(!st.empty()) st.pop();
	a[n+1]=0;
	st.push(n+1);
	for(int i=n;i>=1;i--)
	{
		while(a[i]<a[st.top()])
		{
			st.pop();
		}
		r[i]=st.top();
		st.push(i);
	}
	for(int i=1;i<=n;i++)
	{
		ans-=a[i]*(r[i]-i)*(i-l[i]);
	}
	cout<<ans;
	return 0;
 }