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 intusing 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; }