码迷,mamicode.com
首页 > 其他好文 > 详细

A1118 Birds in Forest (25分)

时间:2020-05-31 01:09:22      阅读:75      评论:0      收藏:0      [点我收藏+]

标签:相同   数字   需要   rds   其他   end   一个   输出   void   

一、技术总结

  1. 题意是,给出N行,每行的第一个数字num,是接下来的鸟的数量。然后,如果两张图片中有相同的鸟的id号,代表这两张照片拍的是同一棵树。最后要求输出在树的数量和鸟的数量。同时给出k对数,查看这两只鸟是否在同一棵树上。
  2. 抽象来理解就是考查并查集,需要将同一棵树上的鸟放进一个集合,关键是理解在合并的操作,我们只要将每行合并操作即可,如果其他行有出现的鸟,那么意味着这两行中的所有鸟会被合并到一个集合中。
  3. 最后需要用一个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\n", 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\n");
		else printf("No\n");
	}
	return 0;
}

A1118 Birds in Forest (25分)

标签:相同   数字   需要   rds   其他   end   一个   输出   void   

原文地址:https://www.cnblogs.com/tsruixi/p/12995641.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!