[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。 对于30%的数据，满足1 ≤ N, M ≤ 1,000。对于60%的数据，满足1 ≤ N, M ≤ 1,000,000。对于100%的数据，满足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。

---

**

Source

**

---

**

Solution

**

刚刚切掉了虔诚的穆主任【墓主人】，起初觉得会很麻烦的样子……就一直不敢写……

发现还是很好写的QAQ

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mod 2147483647
const int maxn=1e5+10;
struct data{int x,y,l,r,u,d;}d[maxn];
int tmp[maxn],tot,n,m,K,W,ans;

bool cmp(data x,data y){return x.y<y.y||(x.y==y.y&&x.x<y.x);}
bool cmp2(data x,data y){return x.x<y.x||(x.x==y.x&&x.y<y.y);}

int sum[maxn*2];
void add(int x,int w)
{
    while(x<=W)
        sum[x]+=w,x+=x&-x;
}
int ask(int x)
{
    int res=0;
    while(x)res+=sum[x],x-=x&-x;
    return res;
}

int C[maxn][11];
int main()
{
    scanf("%d%d%d",&n,&m,&W);
    for(int i=1;i<=W;i++)
        scanf("%d%d",&d[i].x,&d[i].y);
    scanf("%d",&K);
    for(int i=0,j;i<=W;i++)for(j=1,C[i][0]=1;j<=min(i,K);j++)C[i][j]=C[i-1][j-1]+C[i-1][j];
    //离散
    for(int i=1;i<=W;i++)tmp[i]=d[i].x;
    sort(tmp+1,tmp+1+W);
    tot=unique(tmp+1,tmp+1+W)-tmp-1;
    for(int i=1;i<=W;i++)d[i].x=lower_bound(tmp+1,tmp+1+tot,d[i].x)-tmp;

    for(int i=1;i<=W;i++)tmp[i]=d[i].y;
    sort(tmp+1,tmp+1+W);
    tot=unique(tmp+1,tmp+1+W)-tmp-1;
    for(int i=1;i<=W;i++)d[i].y=lower_bound(tmp+1,tmp+1+tot,d[i].y)-tmp;
    //预处理
    sort(d+1,d+1+W,cmp);
    for(int L=1,R=1;L<=R&&R<=W;L=R=R+1)
    {
        while(d[R+1].y==d[R].y)R++;
        for(int i=L;i<=R;i++)
            d[i].l=i-L,d[i].r=R-i;
    }
    sort(d+1,d+1+W,cmp2);
    for(int L=1,R=1;L<=R&&R<=W;L=R=R+1)
    {
        while(d[R+1].x==d[R].x)R++;
        for(int i=L;i<=R;i++)
            d[i].d=i-L,d[i].u=R-i;
    }
    //solve
    sort(d+1,d+1+W,cmp);

    for(int i=1;i<=W;i++)
    {
        add(d[i].x,C[d[i].u][K]*C[d[i].d+1][K]-C[d[i].u+1][K]*C[d[i].d][K]);
        if(d[i+1].y==d[i].y)ans+=(ask(d[i+1].x-1)-ask(d[i].x))*C[d[i].l+1][K]*C[d[i+1].r+1][K];
    }
    printf("%d",ans&mod);
    return 0;
}

---

——呵呵哒QAQ