一贯的链接: http://dev.luogu.org:8888/wiki/show?name=%E9%A2%98%E8%A7%A3+P2154
又是神题!!!!
注意 他说自然溢出! 坑死我了啊 自然溢出会变负数!!! 还要自己加上MO啊!!!
代码
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
#define MO 2147483648
struct H
{
int x,y;
int UP,DOWN; //这个点上面有多少棵树(包括这个点) ,这个点下面有多少棵树(不包括这个点)
int id;
}a[100000+1+1];
bool cmpx(H a,H b){ if(a.x==b.x) return a.y<b.y;return a.x<b.x;}
bool cmpy(H a,H b){ if(a.y==b.y) return a.x<b.x;return a.y<b.y;}
int n,m,w,k;
int s[100000+1],id[100000+1],cnt=0; //树状数组 离散
int c[100000+1][10+1]; //组合
int R[100000+1]; //每行 树的数量
int lowbit(int x){return x&-x;}
int Sum(int x)
{
// cout<<x<<endl;
int ans=0;
while(x>=1)
ans+=s[x],x-=lowbit(x);
return ans;
}
void Add(int x,int d)
{
while(x<=cnt)
s[x]+=d,x+=lowbit(x);
}
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);
a[0].x=a[0].y=a[w+1].x=a[w+1].y=-1;
scanf("%d",&k);
//组合
c[0][0]=1;
for(int i=1;i<=w;i++)
{
c[i][0]=1;
for(int j=1;j<=k;j++)
c[i][j]=c[i-1][j-1]+c[i-1][j];
}
//离散 x id[]
//处理 每颗树 上下 各有多少棵树
sort(a+1,a+w+1,cmpx);
for(int i=1;i<=w;i++) if(a[i].x==a[i-1].x) a[i].DOWN=a[i-1].DOWN+1,a[i].id=cnt; else a[i].id=++cnt;
for(int i=w;i>=1;i--) if(a[i].x==a[i+1].x) a[i].UP=a[i+1].UP+1;else a[i].UP=1;
// 每一行有多少颗树
int tmp=0;
sort(a+1,a+w+1,cmpy);
for(int i=1;i<=w;i++)
{
tmp++;
if(a[i].y!=a[i+1].y) R[i]=tmp,tmp=0;
}
for(int i=w;i>=1;i--) if(a[i].y==a[i+1].y) R[i]=R[i+1];
//求和
tmp=0;
int ans=0;
for(int i=1;i<=w;i++)
{
tmp++;
if(a[i].y!=a[i+1].y)
tmp=0;
else
ans+=(c[tmp][k]*c[ R[i]-tmp ][k])*(Sum(a[i+1].id-1)-Sum(a[i].id));
Add(a[i].id, - c[a[i].UP][k] * c[a[i].DOWN][k] + c[a[i].UP-1][k] * c[a[i].DOWN+1][k] );
}
while(ans<0) ans+=MO; //自然溢出会变负数!!!!!
cout<<ans;
return 0;
}