一、技术总结
- 题意是,给出N行,每行的第一个数字num,是接下来的鸟的数量。然后,如果两张图片中有相同的鸟的id号,代表这两张照片拍的是同一棵树。最后要求输出在树的数量和鸟的数量。同时给出k对数,查看这两只鸟是否在同一棵树上。
- 抽象来理解就是考查并查集,需要将同一棵树上的鸟放进一个集合,关键是理解在合并的操作,我们只要将每行合并操作即可,如果其他行有出现的鸟,那么意味着这两行中的所有鸟会被合并到一个集合中。
- 最后需要用一个bool类型的visit数组,存储出现过的鸟,用于统计鸟的数量,同时设置cnt,我们遍历所有visit中的鸟,如果出现该父节点是新出现的,那么说明出现一棵新树。
二、参考代码
#include<bits/stdc++.h>
using namespace std;
int father[10010];
bool visit[10010] = {false};
set<int> fa;//用于存放父亲结点
int findF(int x){
int a = x;
while(x != father[x]){
x = father[x];
}
while(a != father[a]){
int z = a;
a = father[a];
father[z] = x;
}
return x;
}
void Union(int a, int b){
int faA = findF(a);
int faB = findF(b);
if(faA != faB){
father[faB] = faA;
}
}
int main(){
int n, k, cnt = 0, all_bird = 0;
scanf("%d", &n);
for(int i = 0; i < 10010; i++){
father[i] = i;
}
for(int i = 0; i < n; i++){
int num, id;
scanf("%d %d", &num, &id);
visit[id] = true;
for(int j = 1; j < num;j++){
int index;
scanf("%d", &index);
visit[index] = true;
Union(id, index);
}
}
for(int i = 0; i < 10010; i++){
if(visit[i] && fa.find(findF(i)) == fa.end()){
fa.insert(findF(i));
cnt++;
}
if(visit[i]) all_bird++;
}
printf("%d %d
", cnt, all_bird);
scanf("%d", &k);
for(int i = 0; i < k; i++){
int a, b;
scanf("%d %d", &a, &b);
if(findF(a) == findF(b))printf("Yes
");
else printf("No
");
}
return 0;
}