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。
![](http://acm.hdu.edu.cn/data/images/C20-1007-1.jpg)
![](http://acm.hdu.edu.cn/data/images/C20-1007-1.jpg)
Input
输入包含多组数据,每组数据是一个以0 0结尾的整数对列表,表示了一条通道连接的两个房间的编号。房间的编号至少为1,且不超过100000。每两组数据之间有一个空行。
整个文件以两个-1结尾。
整个文件以两个-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; }