PAT——1114 Family Property 甲级（dfs）

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805356599820288

题意

给定每个人的家庭名称和其名下的房产，要求统计每个家庭（直接或间接相关的成员都属于同一个家庭）成员最小id、成员总数、平均房产数量以及平均房产面积，并按照要求输出

代码解析

采用dfs，dfs过程中将成员放入flag对应位置中，每次dfs结束就代表一个家庭的所有成员已经被找到，flag自增表示开始统计下一个家庭。

两个结构体分别用来存储个人相关信息以及最终要输出的信息，vis用来避免重复

最后注意输出格式即可

AC代码

#include<bits/stdc++.h>
using namespace std;
struct node{
	int num=0,area=0;
};
struct node1{
	int id,num;
	double estate,area;
};
bool cmp(node1 x,node1 y){
	return x.area!=y.area?x.area>y.area:x.id<y.id;
}
unordered_map<int,node> info;
unordered_map<int,vector<int>> fam;
unordered_map<int,int> vis;
vector<int> t,ans[1005];
vector<node1> res;
int flag=0;
void dfs(int root)
{
	vis[root]=1;
	ans[flag].push_back(root);
	for(int i=0;i<fam[root].size();i++)
	{
		if(vis[fam[root][i]]==0)
		{
			dfs(fam[root][i]);
		}
	}
}
int main()
{
	int n,k,x,y,tmp;
	scanf("%d",&n);
	t.resize(n);
	for(int i=0;i<n;i++)
	{
		scanf("%d %d %d",&t[i],&x,&y);
		if(x!=-1) 
		{
			fam[t[i]].push_back(x);
			fam[x].push_back(t[i]);
		}
		if(y!=-1) 
  		{
			fam[t[i]].push_back(y);
			fam[y].push_back(t[i]);	
		}
		scanf("%d",&k);
		while(k--)
		{
			scanf("%d",&tmp);
			fam[t[i]].push_back(tmp);
			fam[tmp].push_back(t[i]);
		}
		scanf("%d %d",&x,&y);
		info[t[i]].num=x;
		info[t[i]].area=y;
	}
	for(int i=0;i<n;i++)
	{
		if(vis[t[i]]==0)
		{
			dfs(t[i]);
			flag++;
		}
	}
	printf("%d\n",flag);
	for(int i=0;i<flag;i++)
	{
		int sum1=0,sum2=0,len=ans[i].size();
		for(int j=0;j<ans[i].size();j++)
		{
			sum1+=info[ans[i][j]].num;
			sum2+=info[ans[i][j]].area;
		}
		res.push_back({*min_element(ans[i].begin(),ans[i].end()),len,sum1*1.0/len,sum2*1.0/len});
	}
	sort(res.begin(),res.end(),cmp);
	for(int i=0;i<flag;i++)
	{
		printf("%04d %d %.3lf %.3lf\n",res[i].id,res[i].num,res[i].estate,res[i].area);
	}
}

注意

注意全局变量和局部变量是否重复，我之前就是局部变量有k，全局变量flag也用的k，结果就出各种错

还有就是include<bits/stdc++.h>这个头文件包含了很多函数，当你定义变量与其重复就会报redeclared或者ambiguous的编译错误。所以如果想图方便引用万能头文件（本人），就要注意变量的命名