并查集 --以cogs259为例

题目链接：http://cogs.pro:8081/cogs/problem/problem.php?pid=pySmxSVgP

【问题描述】

    或许你并不知道，你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱，判断两个人是否是亲戚应该是可行的，但如果两个人的最近公共祖先与他们相隔好几代，使得家谱十分庞大，那么检验亲戚关系实非人力所能及。在这种情况下，最好的帮手是计算机。
    为了将问题简化，你将得到一些亲戚关系的信息，如同Xuebin和Grant是亲戚，Grant和Tension是亲戚等，从这些信息中，你可以推出xuebin和Tension是亲戚。请写一个程序，对于我们的关于亲戚关系的提问，以最快的速度给出答案。

【输入格式】

    输入由两部分组成：
第一部分以N、M开始。N为问题涉及到的人的个数（1＜N＜20000）。这些人的编号为1、2、3、…、N。下面有M行（1＜M＜1000000），每行有两个数ai、bi，表示已知ai和bi是亲戚。
第二部分以Q开始。以下Q行有Q个询问（1＜Q＜1000000），每行为ci、di，表示询问ci和di是否为亲戚。

【输出格式】

输出有若干行，

对于每个询问ci、di，若ci和di为亲戚，则输出Yes，否则输出No。

【输入输出样例】

输入文件

10 7
2 4
5 7
1 3
8 9
1 2
5 6
2 3

3 4
7 10
8 9

输出文件

Yes
No
Yes

并查集：并查集是一种高效的分类算法，当题目中有明显的分类意思时，可考虑用并查集，提取相同元素的特征，进行分类并查找。

第一步：将每个单位作为一个集合。

第二步：查找两个元素，若其集合不同但有相同特征，合并为一个集合。

第三步：按照需求查找对应集合元素。

优化：路径压缩

可知树根的左儿子的儿子和树根右儿子的儿子是一个集合，可将两个孙子都link在树根上，防止形成链式结构，一层一层查找增大时间复杂度。

代码：

 #define _CRT_SECURE_NO_WARNINGS
 #include <iostream>
 #include <cstdio>
 #define MAXN 1000100
 using namespace std;

 int N, M, pre[MAXN] = {  }, Q;
 int ai, bi;

 int find(int x)
 {
     if (x == pre[x])
         return x;
     return pre[x] = find(pre[x]); //路径压缩
 }

 void Union(int x, int y)
 {
     int xPre = find(x);
     int yPre = find(y);
     if (xPre != yPre)
         pre[y] = xPre;
 }

 void preWork()
 {
     for (int i = ; i <= N; i++)
         pre[i] = i;
 }

 int main()
 {
     freopen("relations.in", "r", stdin);
     freopen("relations.out", "w", stdout);
     scanf("%d%d", &N, &M);
     preWork();

     while (M--)
     {
         scanf("%d %d", &ai, &bi);
         Union(ai, bi);
     }
     scanf("%d", &Q);
     while (Q--)
     {
         scanf("%d %d", &ai, &bi);
         int aPre = find(ai);
         int bPre = find(bi);
         if (aPre != bPre)
             printf("No\n");
         else
             printf("Yes\n");
     }
     return ;
 }