合并,查找的集合
第一,建立两个函数,一个用来查找(find),一个用来合并(merge):
如下的一个模板:
题目: 亲戚(Relations)
或许你并不知道,你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥的表姐的孙子。如果能得到完整的家谱,判断两个人是否亲戚应该是可行的,但如果两个人的最近公共祖先与他们相隔好几代,使得家谱十分庞大,那么检验亲戚关系实非人力所能及.在这种情况下,最好的帮手就是计算机。
为了将问题简化,你将得到一些亲戚关系的信息,如同Marry和Tom是亲戚,Tom和B en是亲戚,等等。从这些信息中,你可以推出Marry和Ben是亲戚。请写一个程序,对于我们的关心的亲戚关系的提问,以最快的速度给出答案。
参考输入输出格式 输入由两部分组成。
第一部分以N,M开始。N为问题涉及的人的个数(1 ≤ N ≤ 20000)。这些人的编号为1,2,3,…,N。下面有M行(1 ≤ M ≤ 1000000),每行有两个数ai, bi,表示已知ai和bi是亲戚.
第二部分以Q开始。以下Q行有Q个询问(1 ≤ Q ≤ 1 000 000),每行为ci, di,表示询问ci和di是否为亲戚。
对于每个询问ci, di,若ci和di为亲戚,则输出Yes,否则输出No。
样例输入与输出
输入relation.in
10 7
2 4
5 7
1 3
8 9
1 2
5 6
2 3
3
3 4
7 10
8 9
输出relation.out
Yes
No
Yes
#include<cstdio> #include<algorithm> using namespace std; int ss[20005]; int fin(int x) { if(ss[x] == x) return x; else return fin(ss[x]); //最好用递归,因为用while很容易陷入死循环(这道题就是); /*int r = x; //非递归压缩路径; while(r != id) { int j = ss[r]; ss[r] = id; r = j; } return id;*/ } void merger(int a, int b) //合并函数; { int fa,fb; fa = fin(a); fb = fin(b); if(fa != fb) (fa > fb) ? (ss[fa] = fb) : (ss[fb] = fa); } int main() { int m,n,t; scanf("%d%d%d",&m,&n,&t); for(int i = 1; i <= m; i++) ss[i] = i; //int n,t; //scanf("%d%d",&n,&t); for(int i = 1; i <= n; i++) //并入有关系的亲戚; { int a,b; scanf("%d%d",&a,&b); //printf("find(a)=%d\nfind(b)=%d\n",fin(a),fin(b)); merger(a,b); //printf("1\n"); //ss[fin(a)] = fin(b); } while(t--) { int a1,b1; scanf("%d%d",&a1,&b1); if(fin(a1) != fin(b1)) printf("NO\n"); if(fin(a1) == fin(b1)) printf("YES\n"); } return 0; }