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 ****************************************************/