【bzoj1227】【SDOI2009】【虔诚的墓主人】【组合数学+树状数组】

Description

小W 是一片新造公墓的管理人。公墓可以看成一块N×M 的矩形，矩形的每个格点，要么种着一棵常青树，要么是一块还没有归属的墓地。当地的居民都是非常虔诚的基督徒，他们愿意提前为自己找一块合适墓地。为了体现自己对主的真诚，他们希望自己的墓地拥有着较高的虔诚度。一块墓地的虔诚度是指以这块墓地为中心的十字架的数目。一个十字架可以看成中间是墓地，墓地的正上、正下、正左、正右都有恰好k 棵常青树。小W 希望知道他所管理的这片公墓中所有墓地的虔诚度总和是多少

Input

第一行包含两个用空格分隔的正整数N 和M，表示公墓的宽和长，因此这个矩形公墓共有(N+1) ×(M+1)个格点，左下角的坐标为(0, 0)，右上角的坐标为(N, M)。第二行包含一个正整数W，表示公墓中常青树的个数。第三行起共W 行，每行包含两个用空格分隔的非负整数xi和yi，表示一棵常青树的坐标。输入保证没有两棵常青树拥有相同的坐标。最后一行包含一个正整数k，意义如题目所示。

Output

包含一个非负整数，表示这片公墓中所有墓地的虔诚度总和。为了方便起见，答案对2,147,483,648 取模。

Sample Input

5 6
13
0 2
0 3
1 2
1 3
2 0
2 1
2 4
2 5
2 6
3 2
3 3
4 3
5 2
2

Sample Output

6

HINT

图中，以墓地(2, 2)和(2, 3)为中心的十字架各有3个，即它们的虔诚度均为3。其他墓地的虔诚度为0。

所有数据满足1 ≤ N, M ≤ 1,000,000,000，0 ≤ xi ≤ N，0 ≤ yi ≤ M，1 ≤ W ≤ 100,000， 1 ≤ k ≤ 10。存在50%的数据，满足1 ≤ k ≤ 2。存在25%的数据，满足1 ≤ W ≤ 10000。

注意：”恰好有k颗树“，这里的恰好不是有且只有，而是从>=k的树中恰好选k棵

题解：

          首先离散横纵坐标.

          以纵坐标为第一关键字,横坐标为第二关键字排序依次考虑每个点.

          对于相邻的两个点a,b;

          如果a.y=b.y;设在这一行,a左边算上a有sa棵树,b右边算上b有sb棵树.

          那么这一行上a,b之间的点都会产生c[sa][k]*c[sb][k]*t的贡献.(c[ ][ ]是组合数).

          t是a,b之间的点在它们的列上会产生的贡献.即形如(c[a1][k]*c[b1][k]+c[a2][k]*c[b2][k]+...)

          这个显然可以用树状数组维护.

          现在考虑加入一个点对纵坐标这个位置的影响.可以发现就是下面增加了一个点,上面减少了一个点.

          改变量就是c[s1][k]*c[s2][k]-c[s1+1][k]*c[s2-1][k].直接在树状数组里修改即可.

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100010
#define P 2147483648 
using namespace std;
int n,m,w,q[N<<1],cnt,tot,h[N<<1],l[N<<1],p,s[N],k,fg;
long long ans,t[N<<1],c[N][15];
struct use{int x,y;}a[N];
int find(int x){
  int l=1,r=tot;
  while (l<=r){
  	int mid=(l+r)>>1;
  	if (x<=q[mid]) r=mid-1;
  	else l=mid+1;
  }
  return l;
}
void pre(){
  c[0][0]=1;
  for (int i=1;i<=w;i++){
    c[i][0]=1;
    for (int j=1;j<=min(10,w);j++)
      c[i][j]=(c[i-1][j]+c[i-1][j-1])%P;
  }
}
bool cmp(use a,use b){
  if (a.y==b.y) return a.x<b.x;
  else return a.y<b.y;
}
void add(int x,int v){
  for (int i=x;i<=2*w;i+=i&(-i)) (t[i]+=v)%=P;
}
long long query(int x){
  long long ans(0);
  for (int i=x;i;i-=i&(-i)) ans+=t[i];
  return ans;	
}
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);
    q[++cnt]=a[i].x;q[++cnt]=a[i].y;
  }	
  scanf("%d",&k);
  sort(q+1,q+cnt+1);q[0]=-1;
  for (int i=1;i<=cnt;i++) if (q[i]!=q[i-1]) q[++tot]=q[i];
  for (int i=1;i<=w;i++){h[find(a[i].y)]++;l[find(a[i].x)]++;}
  pre();sort(a+1,a+w+1,cmp);
  for (int i=1;i<=w;i++){
    if (i!=1&&a[i].y==a[i-1].y){
      fg++; 
	  long long s1=c[fg][k]*c[h[find(a[i].y)]-fg][k];s1%=P;
      long long s2=query(find(a[i].x)-1)-query(find(a[i-1].x));
	  (ans+=s1*s2)%=P;
	}
	else fg=0;
    p=find(a[i].x);s[p]++;
    int d=(c[s[p]][k]*c[l[p]-s[p]][k]-c[s[p]-1][k]*c[l[p]-s[p]+1][k])%P;
    add(p,d);
  }
  ans=(ans+P)%P;
  printf("%d\n",ans);
}