Luogu1196【NOI2002】

Luogu1196

这种题目，显然用并查集路径迭代（带权并查集）
就有点像给并查集定向，维护连通块大小与到老祖宗的距离
这样类似差分一样就可以解决很多有关距离、个数的问题。。
具体实现只要自己画个图就很容易推了，一般都这样实现：
（显然无法按秩合并）

int getfa(int x){
    if(fa[x]==x) return x;
    int k=fa[x];
    fa[x]=getfa(fa[x]),D[x]+=D[k];
    return fa[x];
}

这题的话也就很简单了，复杂度$O(T*log_N)$
PS：并查集按秩合并与路径压缩单独使用都是$O(log_N)$的，只有合体时才是$O(α)$

#include<bits/stdc++.h>
#define gt() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++)
#define pt(ch) (Top<1000000?St[Top++]=ch:(fwrite(St,1,1000000,stdout),St[(Top=0)++]=ch))
using namespace std;
int Top;static char St[1000000],buf[1000000],*p1=buf,*p2=buf;
template<typename Tp>inline Tp abs_(Tp x){return x<0?-x:x;}
const int N=(3e4);
int T,n,fa[N+5],cnt[N+5],D[N+5];
int read(){
    int ret=0;char ch=gt();
    while(ch<'0'||ch>'9') ch=gt();
    while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=gt();
    return ret;
}
char red(){char ch=gt();while(!isupper(ch)) ch=gt();return ch;}
void write(int x){if(x>9) write(x/10);pt(x%10+'0');}
int getfa(int x){
    if(fa[x]==x) return x;
    int k=fa[x];
    fa[x]=getfa(fa[x]),D[x]+=D[k];
    return fa[x];
}
void merge(int x,int y){
    x=getfa(x),y=getfa(y);
    if(x!=y) D[x]+=cnt[y],cnt[y]+=cnt[x],cnt[x]=0,fa[x]=y;
}
int main(){
    T=read();
    for(int i=1;i<=N;i++) fa[i]=i,cnt[i]=1;
    while(T--){
        char flg=red();int x=read(),y=read();
        if(flg=='M') merge(x,y);
          else if(getfa(x)==getfa(y)) write(abs_(D[x]-D[y])-1),pt('\n');else pt('-'),pt('1'),pt('\n');
    }
    fwrite(St,1,Top,stdout);
    return 0;
}