HDU——1272小希的迷宫（并查集+拓扑排序）

小希的迷宫

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 41183    Accepted Submission(s): 12691

Problem Description

上次Gardon的迷宫城堡小希玩了很久（见Problem B），现在她也想设计一个迷宫让Gardon来走。但是她设计迷宫的思路不一样，首先她认为所有的通道都应该是双向连通的，就是说如果有一个通道连通了房间A和B，那么既可以通过它从房间A走到房间B，也可以通过它从房间B走到房间A，为了提高难度，小希希望任意两个房间有且仅有一条路径可以相通（除非走了回头路）。小希现在把她的设计图给你，让你帮忙判断她的设计图是否符合她的设计思路。比如下面的例子，前两个是符合条件的，但是最后一个却有两种方法从5到达8。 

 

Input

输入包含多组数据，每组数据是一个以0 0结尾的整数对列表，表示了一条通道连接的两个房间的编号。房间的编号至少为1，且不超过100000。每两组数据之间有一个空行。 
整个文件以两个-1结尾。

 

Output

对于输入的每一组数据，输出仅包括一行。如果该迷宫符合小希的思路，那么输出"Yes"，否则输出"No"。

 

Sample Input

6 8 5 3 5 2 6 4 5 6 0 0 8 1 7 3 6 2 8 9 7 5 7 4 7 8 7 6 0 0 3 8 6 8 6 4 5 3 5 6 5 2 0 0 -1 -1

 

Sample Output

Yes Yes No

 

这题比较坑爹。昨天写了一下本地测试都是错误的。然后不想写了，今天中午理了下思路，很顺地写了下来本地测试直接过，花式自信以为能AC，然后抓狂的WA开始了，各种改各种WA。后来发现光拓扑排序是不够的，两个不相连的迷宫也是可以被拓扑排序的。因此还需要并查集。判断题中任意一点是否属于一个集合。然后又WA了一次才过。并查集果然是黑科技。查询的时候就会自动合并。

代码：

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define MM(x) memset(x,0,sizeof(x))
using namespace std;
typedef long long LL;
const int N=100010;
vector<int>edge[N];
map<int,int>deg;
int pre[N];
int ran[N];
inline int find(int n)
{
	if(n!=pre[n])
		return pre[n]=find(pre[n]);
	return pre[n];
}
inline void joint (int a,int b)
{
	int fa=find(a),fb=find(b);
	if(fa!=fb)
	{
		if(ran[fa]>ran[fb])
		{
			ran[fa]+=ran[fb];
			pre[fb]=fa;
		}
		else
		{
			ran[fb]+=ran[fa];
			pre[fa]=fb;
		}
	}
}
inline void init()
{
	for (int i=0; i<N; i++)
	{
		pre[i]=i;
		ran[i]=1;
	}
}
int main(void)
{
	int x,y,i,j;
	init();
	while (~scanf("%d%d",&x,&y))
	{
		if(x==-1&&x==y)
			break;
		else if(x==0&&y==0)
		{
			map<int,int>::iterator it;
			queue<int> Q;
			set<int>tp;
			for (it=deg.begin(); it!=deg.end(); it++)
			{
				if(it->second<=1)
				{
					it->second--;
					Q.push(it->first);
					tp.insert(it->first);
				}
			}
			while (!Q.empty())
			{
				int now=Q.front();
				Q.pop();
				for (i=0; i<edge[now].size(); i++)
				{
					int v=edge[now][i];
					deg[v]--;
					if(deg[v]==1)
						Q.push(v);
					if(deg[v]==0)
						tp.insert(v);
				}
			}
			bool flag=1;
			int cnt=0,father,now;
			for (it=deg.begin(); it!=deg.end(); it++)
			{
				now=find(it->first);
				if(cnt==0)
				{
					father=find(it->first);
					cnt++;
				}
				else
				{
					if(now!=father)
					{
						flag=0;
						break;
					}	
				}
			}
			if(tp.size()==deg.size()&&flag)
				puts("Yes");
			else
				puts("No");
			deg.clear();
			for (i=0; i<N; i++)
				edge[i].clear();
			init();
		}
		else
		{
			joint(x,y);
			deg[x]++;
			deg[y]++;
			edge[x].push_back(y);
			edge[y].push_back(x);
		}
	}
	return 0;
}