bzoj1227: [SDOI2009]虔诚的墓主人

传送门
先离散横纵坐标
按照y进行排序，从下往上处理每一行
预处理出l[a],r[a],u[a],d[a]表示一个点上下左右的点数。
如果a,b在同一行，则ans+=c（l[a]+1(包括a)，k）*c(r[b]+1,k)再分别乘上ab间的每一个点的c(u[i],k)*c(d[i],k)
但是这样复杂度为n^2，显然是会T掉的。
于是我们要用树状数组维护a到b所有点的c(u[i],k)*c(d[i],k)之和
因为只有一列上出现了一个点，c(u[i],k)*c(d[i],k)才会修改。
所以修改次数为N
可以用树状数组维护实现$O（NlogN）$的时间复杂度

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<cstdlib> 
#define N 100005
#define ll int
using namespace std;
ll lx,ly,n,k,cnt,b[N],c[N][11],num[N],hash[N],sum[N],tmp,ans;
struct data{
    ll x,y,l,r,u,d;
}a[N];
ll read(){
    ll x=0;
    char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar());
    for (;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-48;
    return x;
}
void add(ll x,ll t){
    //printf("%d %d\n",x,t);
    for (;x<=cnt;x+=x&(-x)) b[x]+=t;
}
ll qry(ll x){
    ll t=0;
    for (;x;x-=x&(-x)) t+=b[x];
    return t;
}
ll find(ll x){
    ll l=1,r=cnt,mid;
    while (l<r){
        mid=(l+r)/2;
        if (hash[mid]<x) l=mid+1; else r=mid;
    }
    return l;
}
bool cmp(data a,data b){return a.y<b.y||(a.x<b.x&&a.y==b.y);}
int main(){
    lx=read();
    ly=read();
    n=read();
    for (ll i=1;i<=n;i++){
        num[i]=a[i].x=read();
        a[i].y=read();
    }
    sort(num+1,num+n+1);
    hash[cnt=1]=num[1];
    for (ll i=2;i<=n;i++) 
        if (num[i]!=num[i-1]) hash[++cnt]=num[i];
    sort(a+1,a+n+1,cmp);
    for (ll i=1;i<=n;i++) a[i].x=find(a[i].x);
    tmp=ans=0;
    for (ll i=1;i<=n;i++){
        if (a[i].y==a[i-1].y) tmp++; else tmp=1;
        a[i].l=tmp;
        sum[a[i].x]++;
        a[i].u=sum[a[i].x];
    }
    tmp=0;
    for (ll i=n;i;i--){
        if (a[i].y==a[i+1].y) tmp++; else tmp=1;
        a[i].r=tmp;
        a[i].d=sum[a[i].x]-a[i].u;
    }
    c[0][0]=1; 
    k=read();  
    for (ll i=1;i<=n;i++){  
        c[i][0]=1;  
        for (ll j=1;j<=k&&j<=i;j++) c[i][j]=c[i-1][j]+c[i-1][j-1];  
    }  
    for (ll i=1;i<=n;i++){  
        add(a[i].x,c[a[i].u][k]*c[a[i].d][k]-qry(a[i].x)+qry(a[i].x-1));  
        if (i>1&&a[i].y==a[i-1].y)  
            ans+=c[a[i-1].l][k]*c[a[i].r][k]*(qry(a[i].x-1)-qry(a[i-1].x));  
    }  
    printf("%d\n",ans&2147483647);  
    return 0; 
}