【Ybt OJ】[图论 第1章] 并查集算法 [前半章]

$「$图论$」$第$1$章 并查集$($前$3$题$)$
目录：

A. [模板]并查集
B.程序自动分析
C.银河英雄传说

$A.$ 例题$1$ $[$模板$]$并查集

洛谷$link$

分析：

并查集模板 只需要打出找祖先函数 和合并函数 然后根题面做即可

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue> 
#include<bitset>
//#pragma GCC optimize(2)
#define reg register 
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
const int N=1e4+2;
int n,m,a,b,c,fa[N];
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}  //找祖先函数
void hb(int x,int y)
{
	if(find(x)!=find(y)) fa[find(x)]=find(y);  //合并
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		if(a==1) hb(b,c);
		if(a==2)
		{
			if(find(b)==find(c))
				puts("Y");
			else puts("N");
		}
	}
	return 0;
}

$B.$ 例题$2$ 程序自动分析

洛谷$link$

分析：

并查集 这些判断条件都可以根据并查集的合并情况来看
注意先按$e$排序 从大到小 这样就可以先合并$e=1$的情况 再看$e=0$去判断
然后看到数据范围 有$10^9$那么大 肯定不能把$fa$数组也开得那么大
这样就需要离散化 用$stl$去重 然后二分查找离散化后的位置 就可以辽

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue> 
#include<bitset>
//#pragma GCC optimize(2)
#define reg register 
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
const int N=1e6+5;
int T,n,fa[N],discr[N];
struct node{
	int x,y,kd;
}a[N];
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
bool cmp(node x,node y){return x.kd>y.kd;}
int main(){
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		int tot=0;
		memset(discr,0,sizeof(discr));
		memset(a,0,sizeof(a));
		memset(fa,0,sizeof(fa));
		bool f=1;
		for(int i=1;i<=n;i++)
		{
			scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].kd);
			discr[++tot]=a[i].x;
			discr[++tot]=a[i].y;
		}
		sort(discr+1,discr+tot+1);
		int size=unique(discr+1,discr+tot+1)-discr;  //去重后元素个数
		for(int i=1;i<=n;i++)
		{
			a[i].x=lower_bound(discr+1,discr+size+1,a[i].x)-discr;
			a[i].y=lower_bound(discr+1,discr+size+1,a[i].y)-discr;  //二分查找位置
		}
		for(int i=1;i<=size;i++) fa[i]=i;  //并查集
		sort(a+1,a+n+1,cmp);
		for(int i=1;i<=n;i++)
		{
			int fx=find(a[i].x),fy=find(a[i].y);
			if(a[i].kd==1) fa[fx]=fy;
			else if(a[i].kd==0&&fx==fy){
				puts("NO");
				f=0; break;
			}
		}
		if(f) puts("YES");
	}
	
	return 0;
}

$C.$ 例题$3$ 银河英雄传说

洛谷$link$

分析：

模拟是肯定过不了的 看他让我们判断在不在同一列 以及又让我们合并两队 这样就可以用并查集写
然后就写一个$find$函数来看飞船${}_{i,j}$之间的祖先是不是一个祖先 是就在同一列了 合并两队也是同理 把两队的祖先合并 即飞船${}_i$祖先的祖先赋为飞船${}_j$的祖先
然后求数量 其实就是求距离 就可以用一个前缀和 分别表示飞船${}_i$到队头 飞船${}_j$到队头的距离
这样就可以求出两飞船间的距离了
然后这个前缀和要在找祖先函数中递归 这样才能用正确的祖先求和 再合并队列 和他们的祖先

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue> 
#include<bitset>
//#pragma GCC optimize(2)
#define reg register
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
const int N=3e4+4;
int fa[N],now[N],dis[N],ans,T;
int find(int x){
	if(fa[x]==x) return fa[x];
    int fx=find(fa[x]);
   	dis[x]+=dis[fa[x]];  //求距离
    return fa[x]=fx;
}
int main(){
	for(int i=1;i<=30000;i++)
	{
		fa[i]=i;dis[i]=0;
		now[i]=1;
	}
	scanf("%d",&T);
	while(T--)
	{
		char kd;
		int x,y;
		cin>>kd;
		scanf("%d%d",&x,&y);
		int fx=find(x),fy=find(y);
		if(kd=='C')
		{
			if(fx!=fy) puts("-1");
			else printf("%d\n",abs(dis[x]-dis[y])-1);  //两飞船间距离
		}else if(kd=='M')
		{
			//now[i]表示i为队头的队列长度(飞船数量)
			dis[fx]+=now[fy];  //加上新合并进来的长度
			fa[fx]=fy;
			now[fy]+=now[fx];  //更新
			now[fx]=0;  //不存在了 被合并了
		}
	}
	return 0;
}