题目描述:
给定n个数,定义一个序列的价值为这个序列中最大值减去最小值,问n个数中所有连续的序列价值之和是多少。
做法:
拿到题目边立马想到了边行ans=a[i]*作为最大值的次数-a[i]*作为最小值的次数。
那么怎么求这个次数呢?
刚开始想到了一个n^2的暴力做法,向左边扩展找到第一个小于a[i]的数的位置记为l,然后向右边扩展找到第一个小于或等于a[i]的数的位置记为r。那么次数就是l*r-1。
后面发现这个东西竟然是可以优化的。方法是用单调栈来优化(可见洛谷2947)
代码如下:
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#define N 1000005
using namespace std;
long long a[N],Minl[N],Minr[N],Maxl[N],Maxr[N];
struct Edge{
long long x;
int number;
}s[N];
int main()
{
int n;
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%lld",&a[i]);
memset(s,0,sizeof(s));
int top=1;
Minl[1]=0; s[top].x=a[1];
s[top].number=1;
for (int i=2;i<=n;i++){
while (top&&s[top].x>=a[i]) top--;
if (top==0){
top++;
Minl[i]=0;
s[top].x=a[i]; s[top].number=i;
}
else {
top++;
s[top].x=a[i]; s[top].number=i;
Minl[i]=s[top-1].number;
}
}
memset(s,0,sizeof(s));
top=1;
Minr[n]=n+1; s[top].x=a[n];
s[top].number=n;
for (int i=n-1;i;i--){
while (top&&s[top].x>a[i]) top--;
if (top==0){
top++;
Minr[i]=n+1;
s[top].x=a[i]; s[top].number=i;
}
else {
top++;
s[top].x=a[i]; s[top].number=i;
Minr[i]=s[top-1].number;
}
}
memset(s,0,sizeof(s));
top=1;
Maxl[1]=0; s[top].x=a[1];
s[top].number=1;
for (int i=2;i<=n;i++){
while (top&&s[top].x<=a[i]) top--;
if (top==0){
top++;
Maxl[i]=0;
s[top].x=a[i]; s[top].number=i;
}
else {
top++;
s[top].x=a[i]; s[top].number=i;
Maxl[i]=s[top-1].number;
}
}
memset(s,0,sizeof(s));
top=1;
Maxr[n]=n+1; s[top].x=a[n];
s[top].number=n;
for (int i=n-1;i;i--){
while (top&&s[top].x<a[i]) top--;
if (top==0){
top++;
Maxr[i]=n+1;
s[top].x=a[i]; s[top].number=i;
}
else {
top++;
s[top].x=a[i]; s[top].number=i;
Maxr[i]=s[top-1].number;
}
}
long long Max1=0; long long Min1=0;
for (int i=1;i<=n;i++){
Max1=Max1+a[i]*((Maxr[i]-i)*(i-Maxl[i])-1);
Min1=Min1+a[i]*((Minr[i]-i)*(i-Minl[i])-1);
}
printf("%lld",Max1-Min1);
return 0;
}