Codevs3324 新斯诺克

题目描述 Description

斯诺克又称英式台球，是一种流行的台球运动。在球桌上，台面四角以及两长边中心位置各有一个球洞，使用的球分别为1 个白球，15 个红球和6 个彩球（黄、绿、棕、蓝、粉红、黑）共22个球。

击球顺序为一个红球、一个彩球直到红球全部落袋，然后以黄、绿、棕、蓝、粉红、黑的顺序逐个击球，最后以得分高者为胜。斯诺克的魅力还在于可以打防守球，可以制造一些障碍球使对方无法击打目标球而被扣分。正是因为这样，斯诺克是一项充满神奇的运动。

现在考虑这样一种新斯诺克，设母球（母球即是白球，用于击打其他球）的标号为M，台面上有N 个红球排成一排，每一个红球都有一个标号，他们的标号代表了他们的分数。

现在用母球击打这些红球，一杆击打，如果母球接触到红球，就称为“K 到红球”。我们假设，一次可以击打任意多相邻连续的红球，也可以只击打一个球。并且红球既不会落袋，也不会相互发生碰撞，而只是停留在原处。每次击打时候，要想“K 到红球”，至少要击打一个红球，如果想一次击打多个红球，那么击打的红球必须是依次连续排列的。如果一次“K 到红球”所有红球的标号之和的平均数大于母球的标号M，就获得了一个“连击”。

现在请你计算总共能有多少种“连击”方案。

注意：如果当前有标号为1、2、3 的三种红球，母球标号为0，有如下6 种获得“连击”方案：（ 1）、（ 2）、（ 3）、（ 1，2）、（ 2，3）、（ 1，2，3）

输入描述 Input Description

共有两行。

第一行是N，M (N<=100000，M<=10000) ，N 表示台面上一共有N 个红球，M 表示母球的标号。

第二行是N 个正整数，依次表示台面上N 个红球的标号，所有标号均不超过10000。

输出描述 Output Description

只有一个数，为“连击”的方案总数。

样例输入 Sample Input

4 3

3 7 2 4

样例输出 Sample Output

7

数据范围及提示 Data Size & Hint

请看上面。

 #include<iostream>
 #include<algorithm>
 using namespace std;

 int n,m;
 long long ans;
 int f[],tmp[];

 int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-f;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 void  merge(int left,int mid,int right)
 {
     int i=left,j=mid+,k=left;
     while(i<=mid&&j<=right)
     {
         if(f[i]<f[j])
         {
             tmp[k++]=f[i++];
             ans+=right-j+;
         }
         else
         {
             tmp[k++]=f[j++];
         }
     }
     while(i<=mid) tmp[k++]=f[i++];
     while(j<=right) tmp[k++]=f[j++];
     for(int i=left;i<=right;i++)
         f[i]=tmp[i];
 }

 void Mergesort(int left,int right)
 {
     if(left==right) return;
     int mid=(left+right)/;
     Mergesort(left,mid);
     Mergesort(mid+,right);
     merge(left,mid,right);
 }

 int main()
 {
     n=read();m=read();
     for(int i=;i<=n;i++)
     {
         f[i]=read()-m+f[i-];
     }
     Mergesort(,n);//注意0
     cout<<ans<<endl;
     return ;
 }

树状数组：

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 using namespace std;

 int n,m;
 long long ans;
 int a[];

 struct NODE
 {
     int val,pos;
 }node[];
 int c[];

 int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-f;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 bool cmp(NODE a,NODE b)
 {
     if(a.val!=b.val)
         return a.val<b.val;//<求正序对,>求逆序对
     else return a.pos>b.pos;
 }

 int lowbit(int x)
 {
     return x&(-x);
 }

 void update(int x,int num)
 {
     while(x<=n+)
     {
         c[x]+=num;
         x+=lowbit(x);
     }
 }

 int sum(int x)
 {
     int sum=;
     while(x>)
     {
         sum+=c[x];
         x-=lowbit(x);
     }
     return sum;
 }

 int main()
 {
     n=read();m=read();
     node[]=(NODE){,};
     for(int i=;i<=n+;i++)
     {
         a[i]=a[i-]+read()-m;
         node[i].val=a[i];
         node[i].pos=i;
     }
     sort(node+,node+n+,cmp);
     for(int i=;i<=n+;i++)
     {
         ans+=sum(node[i].pos);
         update(node[i].pos,);
     }
     cout<<ans<<endl;
     return ;
 }