HDU - 1272 并查集

               思路：给定一些无向边，判断是否是一棵树。直接利用并查集判定当前加入的两点，是否已经在同一个集合中，如果在，说明一定不符合只有一条路的原则。当所有情况都满足只有一条路的原则，还应该判断是否所有的点都在同一个集合中。

         坑点：当第一整数对是(x, y)时，应该输出"Yes"

AC代码

#include <cstdio>
#include <cmath>
#include <cctype>
#include <algorithm>
#include <cstring>
#include <utility>
#include <string>
#include <iostream>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000") 
#define eps 1e-10
#define inf 0x3f3f3f3f
#define PI pair<int, int> 
typedef long long LL;
const int maxn = 100000 + 5;
int p[maxn], a[maxn];
int find(int x) {
	return p[x] == x ? x : p[x] = find(p[x]);
} 
bool unionset(int x, int y) {
	int px = find(x), py = find(y);
	if(px != py) {
		p[px] = py;
		return true;
	}
	return false;
}

int main() {
	int x, y, st;
	while(scanf("%d%d", &x, &y) == 2) {
		if(x == -1 && y == -1) break;
		if(x == 0 && y == 0) {
			printf("Yes\n");
			continue;
		}
		for(int i = 1; i < maxn; ++i) p[i] = i; 
		memset(a, 0, sizeof(a));
		st = x;
		int flag = 1;
		do{
			a[x]++, a[y]++;
			if(flag) {
				if(!unionset(x, y)) {
					flag = 0;
				}
			}
			scanf("%d%d", &x, &y);
		}while(x != 0 || y != 0);
		//所有点是否在同一个集合中
		int goal = find(st);
		if(flag) {
			for(int i = 1; i < maxn; ++i) {
				if(a[i] && find(i) != goal) {
					flag = 0;
					break;
				}
			}
		} 
		if(flag) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
} 

如有不当之处欢迎指出！