[NOI2002]银河英雄传说

OJ题号：洛谷1196

思路：加权并查集。

对于每个联通块，记录最后（权值最大）的“战舰”。然后按照加权并查集的基本套路合并。最后输出两艘“战舰”权值之差。

#include<iostream>
#include<algorithm>
const int N=30001;
class UnionFindSet {
    private:
        int anc[N],w[N],end[N];
        int Find(int x) {
            if(x==anc[x]) return x;
            int t=Find(anc[x]);
            w[x]+=w[anc[x]];
            return anc[x]=t;
        }
    public:
        UnionFindSet() {
            for(int i=1;i<N;i++) {
                anc[i]=end[i]=i;
                w[i]=0;
            }
        }
        void Union(int x,int y) {
            int t=Find(x);
            w[t]=1;
            anc[t]=end[Find(y)];
            end[Find(y)]=end[t];
        }
        int Query(int x,int y) {
            return (Find(x)==Find(y))?std::abs(w[x]-w[y])-1:-1;
        }
};
UnionFindSet set;
int main() {
    std::ios_base::sync_with_stdio(false);
    int t;
    std::cin>>t;
    while(t--) {
        char op;
        int x,y;
        std::cin>>op>>x>>y;
        if(op=='M') {
            set.Union(x,y);
        }
        else {
            std::cout<<set.Query(x,y)<<std::endl;
        }
    }
    return 0;
}