带权并查集
对于每一列记录id[i]表示第i号战舰的序号,sum[i]表示第i号战舰所在列的战舰个数
对于每一个合并操作,无法对每一个所在当前列的战舰信息进行更新
所以只对队头战舰的信息进行修改,用类似线段树中懒标记的方法
当要对这一艘战舰进行访问时,对其信息进行修改
注意,在并查集中先要对其列首战舰进行更新,在对当前战舰更新
#include <bits/stdc++.h> using namespace std; const int MAXN=31000; int t,fa[MAXN],id[MAXN],sum[MAXN]; inline int find(int x) { if (fa[x]==x) return fa[x]; int f; f=fa[x]; fa[x]=find(fa[x]);//注意 id[x]+=id[f];//在父亲更新后更新当前 return fa[x]; } int main() { scanf("%d",&t); for (int i=1;i<=30010;i++)//初始化 { sum[i]=1; fa[i]=i; id[i]=0; } for (int i=1;i<=t;i++) { char ch[10]; int a,b; scanf("%s%d%d",ch,&a,&b); if (ch[0]=='M') { if (find(a)!=find(b)) { int ga,gb; ga=find(a); gb=find(b); id[ga]+=sum[gb]; sum[gb]+=sum[ga]; sum[ga]=0; fa[ga]=gb; } } else { if (find(a)!=find(b)) printf("-1\n"); else printf("%d\n",abs(id[b]-id[a])-1); } } }