【bzoj 1227】虔诚的墓主人（树状数组）

传送门biu~
上下左右没有树的墓地必然没有虔诚度，所以将原图离散化成一个最大为$w\times w$的图。
对于每块墓地，令在它上、下、左、右方向常青树的数目分别为$u$，$p$，$l$，$r$。则以这个点为中心的十字架的数目是$C_u^k\times C_p^k\times C_l^k\times C_r^k$。
显然枚举每块墓地，时间复杂度为$O(w^2)$会TLE。那么我们考虑在同一行，即$x$坐标相同的两棵常青树之间的所有墓地，它们的$l$和$r$值是相同的。那么这些墓地的虔诚度之和为$C_l^k\times C_r^k\times \Sigma C_u^k\times\Sigma C_d^k$。采用树状数组来求和。
首先预处理出每棵树的$u$，$p$，$l$，$r$值，按行枚举每棵树。用当前行的树来修改树状数组中$\Sigma C_u^k\times\Sigma C_d^k$的值。每次将两棵树之间的贡献加到答案里。
Ps：因为模数特殊，可以直接用$int$的自然溢出，最后取答案的后31位即可。

#include<bits/stdc++.h>
#define lowbit(x) (x&(-x))
using namespace std;
struct data{int x,y,l,r,u,d;}a[100005];
inline bool cmp(data a,data b){return a.y<b.y || a.y==b.y && a.x<b.x;}
int n,m,w,k,ans,len;
int tree[100005],b[100005],num[100005],c[100005][15];
inline void getC(int k){
    for(int i=0;i<=w;++i){
        c[i][0]=1;
        int go=min(i,k);
        for(int j=1;j<=go;++j)  c[i][j]=c[i-1][j]+c[i-1][j-1];
    }
}
inline void add(int x,int val){
    for(int i=x;i<=len;i+=lowbit(i))    tree[i]+=val;
}
inline int search(int x){
    int re=0;
    for(int i=x;i;i-=lowbit(i))     re+=tree[i];
    return re;
}
int main(){
    scanf("%d%d%d",&n,&m,&w);
    for(int i=1;i<=w;++i)   scanf("%d%d",&a[i].x,&a[i].y),b[i]=a[i].x;
    sort(b+1,b+w+1);    len=unique(b+1,b+w+1)-b-1;
    scanf("%d",&k);getC(k);
    sort(a+1,a+w+1,cmp);
    for(int cnt=0,y=-1,i=1;i<=w;++i){
        a[i].x=lower_bound(b+1,b+len+1,a[i].x)-b;
        if(a[i].y!=y)   y=a[i].y,cnt=0;
        a[i].l=cnt++;
        a[i].d=num[a[i].x]++;
    }
    for(int cnt=0,y=-1,i=w;i>=1;--i){
        if(a[i].y!=y)   y=a[i].y,cnt=0;
        a[i].r=cnt++;
        a[i].u=num[a[i].x]-a[i].d-1;
    }
    for(int i=1;i<=w;++i){
        add(a[i].x,c[a[i].u][k]*c[a[i].d+1][k]-(search(a[i].x)-search(a[i].x-1)));
        if(i>1 && a[i].y==a[i-1].y) ans+=c[a[i-1].l+1][k]*c[a[i].r+1][k]*(search(a[i].x-1)-search(a[i-1].x));
    }
    printf("%d",ans&(~0u>>1));
    return 0;
}