acwing每日一题-2

发表于 更新于 分类于 阅读次数： 本文字数： 968 阅读时长 ≈ 1 分钟
acwing每日一题——连通图

题目

给定一个无向图和其中的所有边，判断这个图是否所有顶点都是连通的。

输入格式

输入包含若干组数据。

每组数据第一行包含两个整数 n 和 m，表示无向图的点和边数。

接下来 m 行，每行包含两个整数 x,y，表示点 x 和点 y 相连。

点的编号从 1 到 n。

图中可能存在重边和自环。

输出格式

每组数据输出一行，一个结果，如果所有顶点都是连通的，输出 YES，否则输出 NO。

数据范围

输入最多包含 10 组数据。
1$\leq$n$\leq$1000,
1$\leq$m$\leq$5000,
1$\leq$x,y$\leq$n

输入样例：

1
2
3
4
5
6
7
4 3
1 2
2 3
3 2
3 2
1 2
2 3

输出

1
2
NO
YES

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#include <iostream>
#pragma GCC optimize(3)
#define N 6000
#define endl '\n'
using namespace std;
int fi[N], root[N];
int find(int x)      //并查集搜索
{
    if (fi[x] != x)
        fi[x] = find(fi[x]);
    return fi[x];
}
void merge(int x, int y)     //合并根节点
{
    int a = find(x);
    int b = find(y);
    if (a != b)
    {
        root[b] += root[a];
        fi[a] = b;
    }
}
int main()
{
    int n, m;
    while (cin >> n >> m)
    {
        for (int i = 1; i <= n; i++)
        {
            fi[i] = i;
            root[i] = 1;
        }
        for (int i = 1; i <= m; i++)
        {
            int x, y;
            cin >> x >> y;
            merge(x, y);
        }
        if (root[find(1)] == n)
            cout << "YES" << endl;
        else
            cout << "NO" << endl;
    }

    return 0;
}