[NOI2002]银河英雄传说 边带权并查集

维护两个东西,深度和大小

get操作:

int get(int x){
    if(x==fa[x]) return x;
    int rt=get(fa[x]);
    dep[x]+=dep[fa[x]];
    return fa[x]=rt; 
}

合并(unite)操作:

void unite(int x,int y){
    x=get(x),y=get(y);
    fa[x]=y;
    dep[x]=siz[y];
    siz[y]+=siz[x];
}

根据深度差就能知道两个结点之间有几个点

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 30010;
int fa[N],n,siz[N],dep[N];
void init(){
	for(int i = 1; i <= 30000; i++) fa[i]=i,siz[i]=1;
}
int get(int x){
	if(x==fa[x]) return x;
	int rt=get(fa[x]);
	dep[x]+=dep[fa[x]];
	return fa[x]=rt; 
}
void unite(int x,int y){
	x=get(x),y=get(y);
	fa[x]=y;
	dep[x]=siz[y];
	siz[y]+=siz[x];
}
int main(){
	scanf("%d",&n);
	init();
	char s[3];
	int a,b;
	for(int i = 1; i <= n; i++){
		scanf("%s%d%d",s,&a,&b);
		if(s[0]=='M'){
			unite(a,b);
		}else{
			if(get(a)!=get(b)) puts("-1");
			else{
				printf("%d\n",abs(dep[a]-dep[b])-1);
			}
		}
	}
	return 0;
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值