acwing每日一题-2

题目

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

输入格式

输入包含若干组数据。

每组数据第一行包含两个整数 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;
}