经典并查集模板题 学习并查集思路 例题 小鑫的城堡sdut OJ 2798

Description
从前有一个国王，他叫小鑫。有一天，他想建一座城堡，于是，设计师给他设计了好多简易图纸，主要是房间的连通的图纸。小鑫希望任意两个房间有且仅有一条路径可以相通。小鑫现在把设计图给你，让你帮忙判断设计图是否符合他的想法。比如下面的例子，第一个是符合条件的，但是，第二个不符合，因为从5到4有两条路径（5-3-4和5-6-4）。

Input
多组输入，每组第一行包含一个整数m（m < 100000），接下来m行，每行两个整数，表示了一条通道连接的两个房间的编号。房间的编号至少为1，且不超过100000。
Output
每组数据输出一行，如果该城堡符合小鑫的想法，那么输出"Yes"，否则输出"No"。
Sample
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

Output

Yes
No

#include <bits/stdc++.h>

using namespace std;

int arr[100000];
int tree[100000];

int find(int x) {
	if (tree[x] == -1) {    //tree[i]==-1 代表这个点是根节点，直接返回这个点即可
		return x;
	}
	else {
		return find(tree[x]);
	}
}

int main() {
	int a;
	while (cin >> a) {
		int num = 0;				//记录图中点的数量
		memset(arr, 0, sizeof(arr));
		memset(tree, -1, sizeof(tree));
		for (int i = 0; i < a; i++) {
			int u, v;
			cin >> u >> v;
			if (arr[u] == 0) {		//如果之前没输入过这个点就计数+1
				num++;
			}
			if (arr[v] == 0) {
				num++;
			}
			arr[u] = arr[v] = 1;
			u = find(u);
			v = find(v);
			if (u != v) {
				tree[u] = v;				//用后者代表集合装起来前边的那个点
			}
		}
		int res = 0;
		for (int i = 0; i < 100000; i++) {
			if (tree[i] == -1 && arr[i]) {	//arr[i]!=0  表示这个点在图上   tree[i]==-1 表示树上这个点是个根节点
				res++;
			}
		}
		if (res == 1 && num - 1 == a)		//如果是一颗树，则  边数==点数-1※
		{
			cout << "Yes\n";
		}
		else {
			cout << "No\n";
		}
	}
	return 0;
}