这道题目帮我回忆了一下逆序对数,用归并排序的方法写的。
如果[i,j]区间内的值的平均值大于a,则sum[j]-sum[i-1]>a*(j-i+1)
即s[j]-a+s[j-1]+a+……+s[i]-a>0;
只需将每个元素减a得到新的前n项和map[j]-map[i-1]>0,j>i
这就是逆序对数了
代码如下:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=100005;
long long sum[maxn];
long long map[maxn];
int n,a;
//用类似归并排序的方法求逆序对数
long long Merge(int begin,int end)
{
long long ans=0;
int mid=begin+end>>1;
int i=begin,j=mid+1,c=begin;
while(i<=mid && j<=end)
{
if(sum[i]>=sum[j]) map[c++]=sum[i++];
else
{
ans+=mid-i+1;
map[c++]=sum[j++];
}
}
while(i<=mid)
map[c++]=sum[i++];
while(j<=end)
map[c++]=sum[j++];
//这句必须要,归并排序的结果
while(begin<=end) sum[begin]=map[begin++];
return ans;
}
long long Mergesort(int begin,int end)
{
if(begin>=end) return 0;
long long res=0;
int mid=begin+end>>1;
if(begin<end)
{
res+=Mergesort(begin,mid);
res+=Mergesort(mid+1,end);
res+=Merge(begin,end);
}
return res;
}
int main()
{
int t,x;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&a);
sum[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&x);
x-=a;
sum[i]=sum[i-1]+x;
}
long long ans=Mergesort(0,n);
printf("%I64d\n",ans);
}
return 0;
}