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