BZOJ 1227: [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棵

解题方法：
好他妈神，树状数组真神。
题解参考：http://blog.csdn.net/lych_cys/article/details/50800916
尊重知识产权，从蒟蒻的我做起

显然有用的坐标只有w*w个，因此先离散化，下面的分析也是针对离散化后的数据。下面的n等同于题目中的w。

考虑一块墓地(i,j)，显然可以O(N^2)求出这个点上下左右的常青树的个数，令为u[i][j],d[i][j],l[i][j],r[i][j]，那么这个墓地为中心的十字架即C(u[i][j],k)*C(d[i][j],k)*C(l[i][j],k)*C(r[i][j],k)。

但是这样还是不行，不妨考虑两个在同一行的相邻常青树中间的墓地，因为它们有相同的l[i][j]和r[i][j]，因此只需要求出中间墓地的ΣC(u[i][j],k)*C(d[i][j],k)即可。由于是同一行，可以将改行一个点的C(u[i][j],k)*C(d[i][j],k)抽象成一个数组，然后维护这个数组，就变成了单点修改前缀查询了。

考虑修改，对于同一列的，只有在跨过一颗常青树之后维护的那个数组对应的值才会变，因此只需要从上到下考虑常青树，那么遍历到某棵常青树后，把它所在列的值修改一下就好了。

//BZOJ 1227

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100005;
int lenx, leny, n, k, cnt;
int c[maxn][11], num[maxn], Hash[maxn], sum[maxn];
int tree[maxn];
struct node{int x, y, l, r, d, u;}a[maxn];
int lowbit(int x){return x&-x;}
void update(int x, int v){for(;x<=cnt;x+=lowbit(x)) tree[x] += v;}
int query(int x){int re=0; for(; x; x-=lowbit(x)) re += tree[x]; return re;}
int getid(int x){
    int 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(node u, node v){
    return u.y < v.y || (u.y == v.y && u.x < v.x);
}
int main()
{
    scanf("%d%d%d", &lenx, &leny, &n);
    for(int i=1; i<=n; i++){
        scanf("%d%d",&a[i].x,&a[i].y);
        num[i]=a[i].x;
    }
    sort(num+1,num+n+1);
    Hash[cnt=1] = num[1];
    for(int i=2; i<= n; i++) if(num[i]!=num[i-1]) Hash[++cnt] = num[i];
    sort(a+1,a+n+1,cmp);
    for(int i=1; i<= n; i++) a[i].x = getid(a[i].x);
    int tmp = 0;
    for(int 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(int 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;
    scanf("%d", &k);
    for(int i=1; i <= n; i++){
        c[i][0] = 1;
        for(int j=1; j<=k&&j<=i; j++){
            c[i][j]=c[i-1][j]+c[i-1][j-1];
        }
    }
    int ans = 0;
    for(int i = 1; i <= n; i++){
        update(a[i].x, c[a[i].u][k]*c[a[i].d][k]-query(a[i].x)+query(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]*(query(a[i].x-1)-query(a[i-1].x));
        }
    }
    printf("%d\n", ans&2147483647);
    return 0;
}