题目难度:三颗星
题目大意:给出一个个家庭的关系和每个人的资产,最后问有多少个家族,并给出每个家族的人数,和资产,以及一个家族成员中最小的编号。
题目坑点:刚开始以为可以用并查集做,但是想了好久都么有想通,今早起来想了想,其实就是图的遍历吧。输入的人员关系,就是两个节点有一条边,理通了思路就好办了,就把关系转换为图,然后对图进行遍历,然后求出有多少个连通子图即可,在遍历的时候顺便记录一些东西就好了。
代码如下:
#include<iostream>
#include<stdlib.h>
#include<queue>
#include<stack>
#include<algorithm>
#include<map>
#include<set>
#include<cstring>
#define MAX 10005
using namespace std;
typedef struct{
int M=0,A=0;
vector<int> rel;
}Node;
typedef struct{
int ID,num;
double avM,avA;
}Rec;
Node G[MAX];
set<int> Sum;
int vis[MAX],totalA,totalM,totalP,minID;
void tie(int a,int b){
G[a].rel.push_back(b);
G[b].rel.push_back(a);
}
void DFS(int n){
vis[n]=1;
totalA+=G[n].A;
totalM+=G[n].M;
totalP++;
if(minID>n)
minID=n;
for(int i=0;i<G[n].rel.size();i++){
if(vis[G[n].rel[i]]==0){
DFS(G[n].rel[i]);
}
}
}
bool cmp(const Rec &a,const Rec &b){
if(a.avA!=b.avA)
return a.avA>b.avA;
else
return a.ID<b.ID;
}
int main(){
fill(vis,vis+MAX,0);
int N,ID,FA,MO,Cnum,Ctemp,Mest,Area;
scanf("%d",&N);
for(int i=0;i<N;i++){
scanf("%d %d %d %d",&ID,&FA,&MO,&Cnum);
Sum.insert(ID);
if(FA!=-1){
tie(FA,ID);
Sum.insert(FA);
}
if(MO!=-1){
tie(MO,ID);
Sum.insert(MO);
}
while(Cnum--){
scanf("%d",&Ctemp);
Sum.insert(Ctemp);
tie(Ctemp,ID);
}
scanf("%d %d",&Mest,&Area);
G[ID].A=Area;
G[ID].M=Mest;
}
vector<Rec> res;
for(set<int>::iterator it=Sum.begin();it!=Sum.end();it++){
int temp=*it;
totalA=0;totalM=0;totalP=0;minID=9999;
if(vis[temp]==0){
DFS(temp);
Rec rec;
rec.ID=minID;
rec.num=totalP;
rec.avA=(double)totalA/totalP;
rec.avM=(double)totalM/totalP;
res.push_back(rec);
}
}
sort(res.begin(),res.end(),cmp);
printf("%d\n",res.size());
for(int i=0;i<res.size();i++){
printf("%04d %d %.3lf %.3lf\n",res[i].ID,res[i].num,res[i].avM,res[i].avA);
}
}