Problem Description
从前有一个国王,他叫小鑫。有一天,他想建一座城堡,于是,设计师给他设计了好多简易图纸,主要是房间的连通的图纸。小鑫希望任意两个房间有且仅有一条路径可以相通。小鑫现在把设计图给你,让你帮忙判断设计图是否符合他的想法。比如下面的例子,第一个是符合条件的,但是,第二个不符合,因为从5到4有两条路径(5-3-4和5-6-4)。
Input
多组输入,每组第一行包含一个整数m(m < 100000),接下来m行,每行两个整数,表示了一条通道连接的两个房间的编号。房间的编号至少为1,且不超过100000。
Output
每组数据输出一行,如果该城堡符合小鑫的想法,那么输出"Yes",否则输出"No"。
Example Input
5 2 5 2 3 1 3 3 6 4 6 6 1 2 1 3 3 4 3 5 5 6 6 4
Example Output
Yes No
Hint
Author
xj
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5 +5;
const int INF = 0x3f3f3f3f;
int pre[maxn], deep[maxn], n[maxn];
int Find(int x) {
return pre[x] == x ? x : Find(pre[x]);
}
void Unite(int x, int y) {
x = Find(x);
y = Find(y);
if(x == y) {
return;
}
if(deep[x] < deep[y]) {
pre[x] = y;
} else {
pre[y] = x;
if(deep[x] == deep[y]) {
deep[x]++;
}
}
}
void CSH(ll k) {
memset(deep, 0, sizeof(deep));
memset(n, 0, sizeof(n));
for(int i = 1; i <= k; i++) {
pre[i] = i;
}
}
void SR(ll m) {
for(int i = 1; i <= m; i++) {
ll x, y;
scanf("%d%d", &x, &y);
n[x] = n[y] = 1;
Unite(x, y);
}
}
void SC(ll m) {
int d = 0, g = 0; //d记录点的个数 g记录根节点个数
for(int i = 1; i <= 2e5; i++) {
if(n[i] == 1) {
d++;
}
if(n[i] == 1 && pre[i] == i) {
g++;
}
}
if(g == 1 && m == d - 1) {
printf("Yes\n");
} else {
printf("No\n");
}
}
int main() {
ll m;
while(~scanf("%lld", &m)) {
CSH(2e5);
SR(m);
SC(m);
}
return 0;
}
/***************************************************
User name:
Result: Accepted
Take time: 108ms
Take Memory: 2544KB
Submit time: 2018-02-01 10:01:48
****************************************************/