连通图（并查集）

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