前言
坚持每天做一道算法题,每天学一点数据结构与算法,写思路和题解来加深自己的印象。
题目描述
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
规定:x 和 y 是亲戚,y 和 z 是亲戚,那么 x 和 z 也是亲戚。如果 x,y 是亲戚,那么 x 的亲戚都是 y 的亲戚,y 的亲戚也都是 x 的亲戚。
输入格式
输出格式
p 行,每行一个 Yes 或 No。表示第 i 个询问的答案为“具有”或“不具有”亲戚关系。
输入输出样例
输入1
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
输出1
Yes
Yes
No
解题思路
并查集,再加上路径压缩。
AC代码
#include <iostream>
using namespace std;
int n, m, p; // 题目输入
int tempa, tempb; // 临时变量
int relative[10000]; // 关系数组
int find(int x) {
if(x == relative[x]) return x;
return relative[x] = find(relative[x]); // 路径压缩
}
int main() {
cin >> n >> m >> p;
for(int i = 0; i < n; i++) relative[i] = i;
for(int i = 0; i < m; i++) {
scanf("%d %d", &tempa, &tempb);
relative[find(tempa)] = find(tempb);
}
for(int i = 0; i < p; i++) {
scanf("%d %d", &tempa, &tempb);
if(find(tempa) == find(tempb)) printf("Yes\n");
else printf("No\n");
}
return 0;
}
总结
其实这题是边学边做的,以前只听说过并查集,但是还不知道怎么用。算法之路很漫长~~