题目链接:
https://begin.lydsy.com/JudgeOnline/problem.php?cid=1318&pid=25
Problem Z: 亲戚
Time Limit: 1 Sec Memory Limit: 128 MB
Description
或许你并不知道,你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱,判断两个人是否亲戚应该是可行的,但如果两个人的最近公共祖先与他们相隔好几代,使得家谱十分庞大,那么检验亲戚关系实非人力所能及。在这种情况下,最好的帮手就是计算机。为了将问题简化,你将得到一些亲戚关系的信息,如Marry和Tom是亲戚,Tom和Ben是亲戚,等等。从这些信息中,你可以推出Marry和Ben是亲戚。请写一个程序,对于我们的关于亲戚关系的提问,以最快的速度给出答案。
Input
输入由两部分组成。
第一部分以N,M开始。N为问题涉及的人的个数(1<=N<=20000)。这些人的编号为1,2,3,…, N。下面有M行(1<=M<=1 000 000),每行有两个数ai, bi,表示已知ai和bi是亲戚。第二部分以Q开始。以下Q行有Q个询问(1<=Q<=1 000 000),每行为ci, di,表示询问ci和di是否为亲戚。
Output
对于每个询问ci, di,输出一行:若ci和di为亲戚,则输出“Yes”,否则输出“No”。
Sample Input
10 7
2 4
5 7
1 3
8 9
1 2
5 6
2 3
3
3 4
7 10
8 9
Sample Output
Yes
No
Yes
HINT
题目分析:
1.引子
暴力做法:爆搜或用逻辑数组来存储,但并不可能过这道题。
所以我们只能采用一种特殊的数据结构:并查集。
2.方法
将每个人抽象化成一个点,数据给出M个边的关系,两个人是亲戚时两点就有一条边。自然就得到了一个N个顶点M条边的图论模型,注意传递关系,在一个图里的一个联通块中任一点都是亲戚。对于最后的Q个提问,即盘短两点是否在同一连通块中。
用集合的思想,对于开始的每一个人都建立一个集合,开始的集合元素就是这个人本身,表示开始不知道任何人是他的亲戚,以后每给出一个亲戚关系就合并两个集合。
对于样例的解释如下:
输入关系 分离集合
初始状态 {1}{2}{3}{4}{5}{6}{7}{8}{9}{10}
(2,4) {1}{2,4}{3}{5}{6}{7}{8}{9}{10}
(5,7) {1}{2,4}{3}{5,7}{6}{8}{9}{10}
(1,3) {1,3}{2,4}{5,7}{6}{8}{9}{10}
(8,9) {1,3}{2,4}{5,7}{6}{8,9}{10}
(1,2) {1,2,3,4}{5,7}{6}{8,9}{10}
(5,6) {1,2,3,4}{5,6,7}{8,9}{10}
(2,3) {1,2,3,4}{5,6,7}{8,9}{10}
由上可知操作是在集合之间的,所以没必要保存边,而且每一步得到的划分方式是动态的。
代码实现:
#include <bits/stdc++.h>
#pragma GCC optimize(2)
using namespace std;
int n,m,q,a,b;
int fa[20001];
void unionn(int x,int y)
{
fa[y]=x; //将y的父亲节点赋值为x
}
/*
int find(int x)
{
if (fa[x]!=x) return find(fa[x]); //若未找到根,及父亲为自己的节点,继续递归
else return x; //直接返回
}
*/
//原来的递归代码
int find(int x)
{
if (fa[x]!=x) fa[x]=find(fa[x]);
//路径压缩,非常关键,在找到根节点时,把路上找到的元素父亲节点指针都指向根节点,所以是fa[x]指向根
//原因:当一棵树退化成一个链时,访问根节点需O(n),会超时
return fa[x]; //返回当前fa[x]指向根后的值,即根
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) fa[i]=i;//预处理,即每个人开始都不是任何人的亲戚
for (int i=0;i<m;i++)
{
scanf("%d%d",&a,&b);
int x=find(a),y=find(b);//找到他们集合里的根节点
if (x!=y) unionn(x,y);//如果他们集合的根节点不同,即不在同一连通块,进行合并
}
scanf("%d",&q);
for (int i=0;i<q;i++)
{
scanf("%d%d",&a,&b);
if (find(a)==find(b)) puts("Yes"); //如果他们的根节点相同,则为亲戚
else puts("No");
}
return 0;
}