10.16T6 逆序对变式

Description

　　斯诺克又称英式台球，是一种流行的台球运动。在球桌上，台面四角以及两长边中心位置各有一个球洞，使用的球分别为1个白球，15个红球和6个彩球（黄、绿、棕、蓝、粉红、黑）共22个球。击球顺序为一个红球、一个彩球直到红球全部落袋，然后以黄、绿、棕、蓝、粉红、黑的顺序逐个击球，最后以得分高者为胜。斯诺克的魅力还在于可以打防守球，可以制造一些障碍球使对方无法击打目标球而被扣分。正是因为这样，斯诺克是一项充满神奇的运动。
　　现在考虑这样一种新斯诺克，设母球（母球即是白球，用于击打其他球）的标号为M，台面上有N个红球排成一排，每一个红球都有一个标号，他们的标号代表了他们的分数。
　　现在用母球击打这些红球，一杆击打，如果母球接触到红球，就称为“K到红球”。我们假设，一次可以击打任意多相邻连续的红球，也可以只击打一个球。
　　并且红球既不会落袋，也不会相互发生碰撞，而只是停留在原处。每次击打时候，要想“K到红球”，至少要击打一个红球，如果想一次击打多个红球，那么击打的红球必须是依次连续排列的。
　　如果一次“K到红球”所有红球的标号之和的平均数大于母球的标号M，就获得了一个“连击”。
　　现在请你计算总共能有多少种“连击”方案。
　　注意：如果当前有标号为1、2、3的三种红球，母球标号为0，有如下6种获得“连击”方：（1）、（2）、（3）、（12）、（2，3）、（1，2，3）

Input

　　输入文件sheeta.in共有两行，第一行是N，M（N<=100000,M<=10000）,N表示台面上一共有N个红球，M表示母球的标号。
　　第二行是N个正整数，依次表示台面上N个红球的标号，所有标号均不超过10000。

Output

　　输出文件sheeta.out只有一个数，为“连击”的方案总数。

Sample Input

4 3

3 7 2 4

Sample Output

7

 

跟官方题解有些出入

由于我们是求平均值，所以我们读入的时候把每一个数字都减去m，然后求前缀和

接着我们的任务就是求多少个区间值是正数

也就是sum[j]>sum[i]同时 j>i 显然就是一个逆序对

鉴于不想写归并排序，我就离散化了一下解决了

最后被long long略卡了一下

code：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define N 1000005
long long c[N+1],sum[N],n,m,a[N]; 
long long lowbit(long long x){
    return x&(-x);
}
void add(long long x,long long v){
    while(x<=N){
        c[x]+=v;
        x+=lowbit(x);
    }
}
long long query(long long x){
    long long ans=0;
    while(x){
        ans+=c[x];
        x-=lowbit(x); 
    }
    return ans;
}
long long d[N],cnt;
int main(){
    cin>>n>>m;
    for(long long i=1;i<=n;i++){
        cin>>a[i];
        a[i]-=m;
        sum[i]=sum[i-1]+a[i];
        d[++cnt]=sum[i];
    }
    d[++cnt]=0;
    sort(d+1,d+cnt+1);
    for(long long i=1;i<=n;i++){
        sum[i]=lower_bound(d+1,d+cnt+1,sum[i])-d;
    //    cout<<sum[i]<<" ";
    } 
    sum[0]=lower_bound(d+1,d+cnt+1,0)-d;
    long long ans=0;
    for(long long i=0;i<=n;i++){
        ans+=query(sum[i]);
        add(sum[i]+1,1);
    }
    cout<<ans;
    return 0;
}

over