[BZOJ1227][SDOI2009]虔誠的墓主人(樹狀陣列+掃描線)
阿新 • • 發佈:2019-01-06
題目:
題解:
首先離散橫縱座標.
以縱座標為第一關鍵字,橫座標為第二關鍵字排序依次考慮每個點.
對於相鄰的兩個點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);
}