int find(int x) //找根结点(找教主)
{
if(pre[x] != find(x)) pre[x] = find(pre[x]);
return pre[x];
}
例题:并查集
给定一个无向图和其中的所有边,判断这个图是否所有顶点都是连通的。
输入格式
输入包含若干组数据。
每组数据第一行包含两个整数 nn 和 mm,表示无向图的点和边数。
接下来 mm 行,每行包含两个整数 x,yx,y,表示点 xx 和点 yy 相连。
点的编号从 11 到 nn。
图中可能存在重边和自环。
输出格式
每组数据输出一行,一个结果,如果所有顶点都是连通的,输出
YES
,否则输出NO
。数据范围
输入最多包含 1010 组数据。
1≤n≤10001≤n≤1000,
1≤m≤50001≤m≤5000,
1≤x,y≤n1≤x,y≤n输入样例:
4 3 1 2 2 3 3 2 3 2 1 2 2 3
输出样例:
NO YES
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1010;
int pre[N];
int find(int x)
{
if(pre[x]!=x) pre[x]=find(pre[x]);
return pre[x];
}
int main()
{
int n,m,a,b;
while(cin>>n>>m)
{
for(int i=1;i<=n;i++) pre[i]=i;
while(m--)
{
cin>>a>>b;
pre[find(a)]=find(b);
}
bool res=true;
for(int i=2;i<=n;i++)
{
if(find(i)!=find(1))
res=false;
}
if(res) puts("YES");
else puts("NO");
}
return 0;
}
2022/7/11