1. 程式人生 > >[BZOJ1227][SDOI2009]虔誠的墓主人(樹狀陣列+掃描線)

[BZOJ1227][SDOI2009]虔誠的墓主人(樹狀陣列+掃描線)

題目:

我是超連結

題解:

首先離散橫縱座標.
以縱座標為第一關鍵字,橫座標為第二關鍵字排序依次考慮每個點.
對於相鄰的兩個點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].直接在樹狀數組裡修改即可.

這裡對於2147483648取模可以有黑科技,只需要在最後&2147483647就行了,中間會自然溢位實際上就是取模了

程式碼:

#include <map>
#include <cstdio>
#include <algorithm>
#define IN unsigned int
using namespace std;
const int N=100005;
struct hh{int x,y,id;}t[N];
int w,n,m,k,l[N],r[N],u[N],d[N],ls[N],sb[N],maxx;
IN c[N][15],ans,v[N];
void
add(int loc,int vv){for (int i=loc;i<=maxx;i+=i&(-i)) v[i]+=vv;} IN qurry(int loc) { IN ans=0; for (int i=loc;i>=1;i-=i&(-i)) ans+=v[i]; return ans; } int cmpx(hh a,hh b){return a.x<b.x || (a.x==b.x && a.y<b.y);} int cmpy(hh a,hh b){return a.y<b.y || (a.y==b.y && a.x<b.x);} void
init() { sort(t+1,t+w+1,cmpx); for (int i=1;i<=w;i++)//u,d(不包含本身 { int j=i,all=0; while (t[j].x==t[j+1].x) j++,d[t[j].id]=++all; while (i<=j) u[t[i].id]=all--,i++; i--; } sort(t+1,t+w+1,cmpy); for (int i=1;i<=w;i++)//l,r(不包含本身 { int j=i,all=0; while (t[j].y==t[j+1].y) j++,l[t[j].id]=++all; while (i<=j) r[t[i].id]=all--,i++; i--; } c[0][0]=1; for (int i=1;i<=w;i++) { c[i][0]=1; for (int j=1;j<=min(i,k);j++) c[i][j]=c[i-1][j-1]+c[i-1][j]; } } int main() { scanf("%d%d",&n,&m);n++;m++; scanf("%d",&w); for (int i=1;i<=w;i++) t[i].id=i,scanf("%d%d",&t[i].x,&t[i].y),t[i].x++,sb[i]=t[i].x,t[i].y++; sort(sb+1,sb+w+1);int s=unique(sb+1,sb+w+1)-sb-1; scanf("%d",&k);init(); for (int i=1;i<=w;i++) ls[i]=lower_bound(sb+1,sb+s+1,t[i].x)-sb,maxx=max(maxx,ls[i]); for (int i=1,j;i<=w;i=j+1)//按y排序 { j=i; add(ls[i],c[d[t[i].id]+1][k]*c[u[t[i].id]][k]-c[d[t[i].id]][k]*c[u[t[i].id]+1][k]); while (t[j].y==t[j+1].y) { j++; ans+=c[l[t[j-1].id]+1][k]*c[r[t[j].id]+1][k]*(qurry(ls[j]-1)-qurry(ls[j-1])); add(ls[j],c[d[t[j].id]+1][k]*c[u[t[j].id]][k]-c[d[t[j].id]][k]*c[u[t[j].id]+1][k]); } } printf("%d",ans&2147483647); }