hdu 2443 Counter Strike

这道题目帮我回忆了一下逆序对数，用归并排序的方法写的。

如果[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;
}