PAT1114. Family Property (25) 关系连接题，DFS遍历应用

最新推荐文章于 2022-04-27 09:14:42 发布

DrawnBreak

最新推荐文章于 2022-04-27 09:14:42 发布

阅读量210

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT

本文链接：https://blog.youkuaiyun.com/a799581229/article/details/79171665

PAT 专栏收录该内容

126 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）遍历宗族关系图的方法，旨在统计宗族数量及成员的平均房产数量与面积。通过构建图模型，并采用DFS算法遍历每个节点，最终实现对宗族信息的有效统计。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/*************************
题意:
给出1个人的id
并给出他的父母id，和他的儿女id
问你最终统计下来，总共有多少个有血缘关系的宗族存在
并统计这些宗族中的人均房产数量，和人均房产面积。
************************/
/***********************
解题思路：
第一反应是并查集，但因为一次统计有多个关系连接存在
如果用并查集非常难确定father值
故想到用dfs图的遍历
即这实际上是个图
对一个点做dfs遍历，便能将这个点所连通的其他点都遍历掉
遍历结束后，就得到了这个宗族的信息

接着再对其他还未被遍历过的点做遍历。
时间复杂度是O（n），因为每个点只需要遍历一次即可
*************************/
/***********************

*********************/
#include<iostream>
#include<stdio.h>
#include<string>
#include<vector>
#include<queue>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
using namespace std;
#define M 10005
#define INF 0x7ffffff

vector<int> edge[M];
int Enum[M];
double Area[M];
int flag[M];

int Fmnum,Fenum,minid;
double Farea;
struct Ans{
	int id,num;
	double Eavg,Aavg;
};
void dfs(int id){
	int e, i ;

	flag[id] = 1;
	Fmnum += 1;
	Fenum += Enum[id];
	Farea += Area[id];
	if(id < minid)
		minid = id;


	for(i = 0;i<edge[id].size();i++){
		e = edge[id][i];
		if(flag[e] == 0){
			dfs(e);
		}
	}
}
bool cmp(Ans a,Ans b){
	if(a.Aavg > b.Aavg)
		return true;
	else if(a.Aavg == b.Aavg){
		return a.id < b.id;
	}
	else return false;
}
int main(){
	int n;
	cin>>n;
	int id,mid,fid,ck,c;
	vector<int> vc;
	vector<int> personv;

	memset(Enum,0,sizeof(Enum));
	for(int i = 0; i <M;i++)
		Area[i] = 0;

	while(n--){
		cin>>id;
		cin>>mid>>fid;
		if(mid != -1){
			edge[id].push_back(mid);
			edge[mid].push_back(id);
		}
		if(fid != -1){
			edge[id].push_back(fid);
			edge[fid].push_back(id);
		}
		cin>>ck;
		vc.clear();
		while(ck--){
			cin>>c;
			edge[id].push_back(c);
			edge[c].push_back(id);
		}
		cin>>Enum[id]>>Area[id];
		personv.push_back(id);
	}

	int i;
	memset(flag, 0 ,sizeof(flag));
	vector<Ans> ans;
	Ans a;
	for(i = 0;i<personv.size();i++){
		id = personv[i];
		if(flag[id] == 0){
			minid = INF;
			Farea = 0;
			Fenum = Fmnum = 0;
			dfs(id);
			a.id = minid;
			a.num = Fmnum;
			a.Aavg = Farea/Fmnum;
			a.Eavg = double(Fenum)/Fmnum;
			ans.push_back(a);
		}
	}
	sort(ans.begin(),ans.end(),cmp);
	cout<<ans.size()<<endl;
	for(i = 0;i<ans.size();i++){
		printf("%04d %d %.3lf %.3lf\n",ans[i].id,ans[i].num,ans[i].Eavg,ans[i].Aavg);
	}
	return 0;
}