传送
时间限制:1s
内存限制:256MB
问题描述
给出一个随机的排列,请你计算最大值减最小值的差小于等于0~n-1的区间分别有多少个。
输入格式
输入文件名为sum.in。
第一行一个数T(<=10),表示数据组数
对于每一组数据:
第一行一个数n(1<=n<=100,000)
第二行n个数a1…an,表示一个随机的排列
输出
输出文件名为sum.out。
对于每组数据输出n行,分别表示差值小于等于0~n-1的区间个数
数据说明
对于30%的数据,1<=n<=300;
对于60%的数据,1<=n<=5000
对于100%的数据,1<=n<=100000
60分n^2的暴力(蒟蒻只想到这里啦)
100分,用单调队列来优化60分的做法。
数据范围很明显提示复杂度要O(nlogn),考虑怎么优化60分的算法.我们向右跳并不需要每次只跳一位,因为可能有很多位置不会更新minx和maxx,我们需要找到下一位能更新minx和maxx的位置,利用单调栈来记录.维护一个单调上升的单调栈和一个单调下降的.因为这是一个随机的排列,所以差不多会跳logn次.
60分
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#define LL long long
#define N 100009
using namespace std;
int a[N],T,n;
LL ans[N];
int main()
{
scanf("%d",&T);
while(T--)
{
memset(ans,0,sizeof(ans));
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
int maxn,minn;
for(int i=1;i<=n;i++)
{
maxn=a[i],minn=a[i];
for(int j=i;j<=n;j++)
{
minn=min(minn,a[j]);
maxn=max(maxn,a[j]);
ans[maxn-minn]++;
}
}
for(int i=0;i<n;i++) printf("%lld ",ans[i]);puts("");
for(int i=1;i<n;i++) ans[i]+=ans[i-1];
for(int i=0;i<n;i++) printf("%lld\n",ans[i]);
}
return 0;
}
100分
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#define LL long long
#define N 100009
using namespace std;
int a[N],T,n;
int q[N],r,pos[N],nxtmax[N],nxtmin[N];
LL ans[N];
int main()
{
scanf("%d",&T);
while(T--)
{
memset(a,0,sizeof(a));
memset(ans,0,sizeof(ans));
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n;i++) nxtmin[i]=nxtmax[i]=n+1;
r=0;
memset(q,0,sizeof(q));
memset(pos,0,sizeof(pos));
for(int i=1;i<=n;i++)
{
while(r>0&&q[r]>a[i])
{
nxtmin[a[pos[r]]]=i;
r--;
}
q[++r]=a[i];
pos[r]=i;
}
//for(int i=1;i<=r;i++) printf("%d ",q[i]);puts("");
r=0;
memset(pos,0,sizeof(pos));
memset(q,127/3,sizeof(q));
for(int i=1;i<=n;i++)
{
while(r>0&&q[r]<a[i])
{
nxtmax[a[pos[r]]]=i;
r--;
}
q[++r]=a[i];
pos[r]=i;
}
//for(int i=1;i<=r;i++) printf("%d ",q[i]);puts("");
for(int i=1;i<=n;i++)
{
int j=i,minn=a[i],maxn=a[i];
while(j<=n)
{
if(nxtmax[maxn]<=nxtmin[minn])
{
ans[maxn-minn]+=nxtmax[maxn]-j;
j=nxtmax[maxn];
maxn=a[nxtmax[maxn]];
}
else
{
ans[maxn-minn]+=nxtmin[minn]-j;
j=nxtmin[minn];
minn=a[nxtmin[minn]];
}
}
}
//for(int i=1;i<=n;i++) printf("%d %d\n",nxtmax[a[i]],nxtmin[a[i]]);
for(int i=1;i<n;i++) ans[i]+=ans[i-1];
for(int i=0;i<n;i++) printf("%lld\n",ans[i]);
}
return 0;
}
// 3 2 4 1