Codevs3324 新斯诺克

• 题目大意：给定一个具有n个数的数列和一个数m，求有多少个连续子序列的和的平均数大于m。

• 思路：平均数是不用管的，数列中每个数直接减去即可，这样条件转化为连续子序列的和大于0。再由此联想到数组的前缀和，则某一段序列和a[i]+…+a[j]=s[j]-s[i-1]，符合条件的话，则有s[j]-s[i-1]>0，即s[j]>s[i-1]，由此转化为求逆序对数。

• 代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=100005;
long long n,m,s[maxn],res=0,c[maxn];

void init()
{
     memset(s,0,sizeof(s));
     scanf("%d%d",&n,&m);
     for (int i=1;i<=n;++i)
     {
         int tmp;
         scanf("%d",&tmp);
         s[i]=s[i-1]+tmp-m;
     }
}

void msort(int l,int r)
{
     if (l==r) 
     {
        c[l]=s[l];
        return;
     }
     int mid=(l+r)/2;
     msort(l,mid);
     msort(mid+1,r);
     int i=l,j=mid+1,k=l;
     while (i<=mid && j<=r)
     {
           if (s[i]>=s[j])
           {
               c[k]=s[j];
               res+=(mid-i+1);
               j++;
               k++;
           }
           else
           {
               c[k]=s[i];
               i++;
               k++;
           }
     }
     while (i<=mid)
     {
           c[k]=s[i];
           i++;
           k++;
     }
     while (j<=r)
     {
           c[k]=s[j];
           j++;
           k++;
     }
     for (int i=l;i<=r;++i)
       s[i]=c[i];
}

int main()
{
    init();
    msort(0,n);
    n++;
    long long ans=n*(n-1)/2;
    printf("%lld",ans-res);
    return 0;
}