BZOJ4298 : [ONTAK2015]Bajtocja
设f[i][j]为第i张图中j点所在连通块的编号,加边时可以通过启发式合并在$O(dn\log n)$的时间内维护出来。
对于每个点,设h[i]为f[j][i]的hash值,若两个点hash值相等,则它们在d张图中均连通。
#include<cstdio> typedef unsigned long long ll; const int D=200,N=5002,M=262143; int d,n,m,i,j,x,y,z,ans; int f[D][N],s[D][N],g[D][N],v[D*N*2],nxt[D*N*2],ed; ll pow[D],h[N]; struct E{ll v;int w,nxt;}e[N]; int G[M+1],res[N],cur; inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';} inline void add(int z,int x,int y){v[++ed]=y;nxt[ed]=g[z][x];g[z][x]=ed;} inline void ins(ll v){ int u=v&M,i=G[u]; for(;i;i=e[i].nxt)if(e[i].v==v){ans+=e[i].w*2+1;e[i].w++;return;} ans++,e[i=res[cur--]].v=v,e[i].w=1,e[i].nxt=G[u],G[u]=i; } inline void del(ll v){ int u=v&M,i=G[u],j=i; if(e[i].v==v){ ans-=e[i].w*2-1; if(!(--e[i].w))G[u]=e[res[++cur]=i].nxt; return; } for(i=e[i].nxt;i;j=i,i=e[i].nxt)if(e[i].v==v){ ans-=e[i].w*2-1; if(!(--e[i].w))e[j].nxt=e[res[++cur]=i].nxt; return; } } void dfs(int z,int x,int y,int t){ del(h[x]); h[x]-=pow[z]*f[z][x]; f[z][x]=t; ins(h[x]+=pow[z]*t); for(int i=g[z][x];i;i=nxt[i])if(v[i]!=y)dfs(z,v[i],x,t); } inline void merge(int z,int x,int y){ if(f[z][x]==f[z][y])return; if(s[z][f[z][x]]>s[z][f[z][y]]){int t=x;x=y;y=t;} s[z][f[z][y]]+=s[z][f[z][x]]; add(z,x,y),add(z,y,x); dfs(z,x,y,f[z][y]); } int main(){ read(d),read(n),read(m); for(pow[0]=i=1;i<d;i++)pow[i]=pow[i-1]*10007; for(i=1;i<=n;i++)res[++cur]=i; for(i=1;i<=n;ins(h[i++]))for(j=0;j<d;j++)f[j][i]=i,s[j][i]=1,h[i]+=pow[j]*i; while(m--)read(x),read(y),read(z),merge(--z,x,y),printf("%d\n",ans); return 0; }