luogu 2154 离散化+杨辉三角+树状数组

将纵向固定，每次在横向找两个点，计算其中间墓地的贡献答案，离散化后同一行的预处理个数，

树状数组内存储C[up[i]][k] * C[down[i][k] 的值，每次更新时 down[横坐标]++;

c[up[i]][k]*C[down[i]][k] - C[up[i]+1][k]*C[down[i]-1][k] 加入这个值即可

左右答案利用 lf(左侧点数)计算

详见代码

#include<bits/stdc++.h>
#define int long long
#define rep(i,x,y) for(register int i=x;i<=y;i++)
#define dec(i,x,y) for(register int i=x;i>=y;i--)
#define mod 2147483648LL
using namespace std;
const int W=;
inline int read(){
    int x=,f=;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-;ch=getchar();}
    while(isdigit(ch)){x=(x<<)+(x<<)+(ch^);ch=getchar();}
    return x*f;}int n,m,w,k,h[*W],allheng[*W],allshu[*W],down[*W],c[*W],C[W][],ans;
inline void change(int x,int d){for(int i=x;i<=*w;i+=i&(-i)) c[i]=(c[i]+d)%mod;return;}
inline int query(int x){int res=;for(int i=x;i;i-=i&(-i)) res=(res+c[i])%mod;return res;}
struct node{int x,y;}a[W];bool cmp(node a,node b){if(a.y==b.y)return a.x<b.x;return a.y<b.y;}
inline void getc(){C[][]=; rep(i,,w){C[i][]=;
    rep(j,,min(i,k)) C[i][j]=(C[i-][j]+C[i-][j-])%mod;}}
inline int find(int x){
    int l=,r=*w;while(l<=r){ int mid=(l+r)>>;
    if(h[mid]<x) l=mid+;else if(x<h[mid]) r=mid-;else return mid;}}
#undef int
int main(){
#define int long long
    //预处理和读入
    n=read(),m=read(),w=read();
    rep(i,,w) a[i].x=read(),a[i].y=read(),h[*i-]=a[i].x,h[*i]=a[i].y;
    k=read(); getc(); sort(h+,h++*w); sort(a+,a++w,cmp);
    rep(i,,w) allshu[find(a[i].x)]++,allheng[find(a[i].y)]++; int lf;

    rep(i,,w){
        if(<i&&a[i-].y==a[i].y){ lf++;//左侧的点数
            int t1=query(find(a[i].x)-)-query(find(a[i-].x));//纵向 区间内答案之和
            int t2=C[lf][k]*C[allheng[find(a[i].y)]-lf][k];//横向
            ans=(ans+t1*t2)%mod;}
        else lf=;

        int ii=find(a[i].x);down[ii]++; //已经以乘积的形式放入树状数组
        change(ii,(C[down[ii]][k]*C[allshu[ii]-down[ii]][k]-C[down[ii]-][k]*C[allshu[ii]-down[ii]+][k])%mod);}
    if(ans<) ans+=mod;printf("%lld\n",ans);return ;}