\(N \times M\)的矩形,格点是共\(W\)棵常青树或墓地。对于一块墓地,它的虔诚度为让它正上下左右各恰有\(k\)棵常青树的方法数量。求出整个矩形公墓的虔诚度总和。
对于 \(30\%\) 的数据,满足 \(1 ≤ N, M ≤ 10^3\)。
对于 \(60\%\) 的数据,满足 \(1 ≤ N, M ≤ 10^6\)。
对于 \(100\%\) 的数据,满足 \(1 ≤ N, M ≤ 10^9\),\(0 ≤ x_i ≤ N\),\(0 ≤ y_i ≤ M\),\(1 ≤ W ≤ 10^5\),\(1 ≤ k ≤ 10\)。
存在 \(50\%\) 的数据,满足 \(1 ≤ k ≤ 2\)。
存在 \(25\%\) 的数据,满足 \(1 ≤ W ≤ 10^4\)。
\(W\)和\(k\)的数据范围较小,考虑在它们身上做文章。
对于\(W\):离散化,此后最多有\(2*W\)个坐标。
对于\(k\)相关的统计:预处理时间不够,考虑边算边统计。
再整理一遍思路:
按y坐标离散化,并统计每一列一共有多少常青树//,并按列扔进树状数组里
按x坐标离散化,并统计每一行一共有多少常青树
当需要计算时,这时扫描常青树的顺序恰好是按行从左到右的顺序
对于相邻两棵常青树,很容易知道它们之间的空地(墓地)都对应着哪一些列,以及这一些列对于这一块墓地的\(C_{up}^{k} \times C_{down}^{k}\)。
这个地方是不是得优化……
每次扫描到一棵树,就可以更新下一行这一列的树的个数了。\(lst_{横坐标}++\)后,把原数\(a\)更新成\(a*(x+1)(y-x-k-1)/(x+1-k)(y-x-1)\).lst
记录到现在(包括现在)每一列有多少树。
还要优化,因为没法开1e9的数组。
离散化之后的坐标也要用上。因此,如果直接乘会出问题。加上我们知道每个可能成为十字架的列的前置常青树情况,所以可以通过加减来改变。
在lst[列号]++之前:
给这个坐标加上\(c[lst[ lh ]+1][k]*c[tot[列号]-lst[列号]-2]-c[lst[列号][k]*c[tot[列号]-lst[列号]-1]\).
树状数组:单点修改,区间查询。
AC 1.43s
//n m w x y k #include<bits/stdc++.h> using namespace std; #define ll long long const int N=1e5+5; const ll M=2147483648; ll n,m; struct evergreentree { ll x,y; #define x(i) et[i].x #define y(i) et[i].y } et[N]; int w,k; ll lst[N],sx[N],sy[N],zh=1,zl=1;//每一行常青树的总数/列,总行数,总列数 ll c[N][15],tr[N],ans=0; inline int lowbit(int i){ return i&(-i); } inline void ad(int p,ll chg){ while(p<N){ tr[p]=(tr[p]+chg)%M; p+=lowbit(p); } } inline ll qs(int i){ ll ret=0; while(i){ ret=(ret+tr[i])%M; i-=lowbit(i); } return ret; } inline bool cmpx(evergreentree xx, evergreentree yy) { if(xx.x==yy.x) return xx.y<yy.y; return xx.x<yy.x; } inline bool cmpy(evergreentree xx, evergreentree yy) { if(xx.y==yy.y) return xx.x<yy.x; return xx.y<yy.y; } inline void ini(){ c[0][0]=1; for(int i=1;i<N;i++){ c[i][0]=1; for(int j=1;j<15;j++) c[i][j]=(c[i-1][j]+c[i-1][j-1])%M; } } int main() { // freopen("1.in","r",stdin); // ini(); scanf("%lld%lld%d",&n,&m,&w); for(int i=1; i<=w; i++) scanf("%lld%lld",&et[i].x,&et[i].y); scanf("%d",&k); sort(et+1,et+w+1,cmpy); for(int i=2; i<=w; i++) { if(y(i-1)!=y(i)) { y(i-1)=zl, sy[zl]++, ++zl; continue; } y(i-1)=zl, sy[zl]++; } y(w)=zl, sy[zl]++; sort(et+1,et+w+1,cmpx); for(int i=2; i<=w; i++) { // for(int i=1;i<=w;i++) printf("%d ",et[i-1].x); // printf("\n"); // printf("%d %d\n",x(i),y(i)); if(x(i-1)!=x(i)) { x(i-1)=zh, sx[zh]++, ++zh; continue; } x(i-1)=zh, sx[zh]++; } x(w)=zh, sx[zh]++; ll p=0; for(int i=1; i<=w; i++) { if(i>1 && x(i-1)==x(i)){ p++; ll m1=c[p][k]%M*c[sx[x(i)]-p][k]%M; ll m2=(qs(y(i)-1)-qs(y(i-1))+M)%M;//? ans+=m1*m2; ans%=M; } else p=0; lst[y(i)]++; ll chg=(c[lst[y(i)]][k]*c[sy[y(i)]-lst[y(i)]][k]-c[lst[y(i)]-1][k]*c[sy[y(i)]-lst[y(i)]+1][k])%M; ad(y(i),chg); } printf("%lld",(ans+M)%M); return 0; } /* 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 6 */