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