L2-007 家庭房產【DFS】

                                                 L2-007 家庭房產

                 https://pintia.cn/problem-sets/994805046380707840/problems/994805068539215872

 

 

題意

給定每個人的家庭成員和其自己名下的房產，請你統計出每個家庭的人口數、人均房產面積及房產套數。

輸入

輸入第一行給出一個正整數N（≤1000），隨後N行，每行按下列格式給出一個人的房產：

編號 父 母 k 孩子1 ... 孩子k 房產套數 總面積

其中編號是每個人獨有的一個4位數的編號；父和母分別是該編號對應的這個人的父母的編號（如果已經過世，則顯示-1）；k（0≤k≤5）是該人的子女的個數；孩子i是其子女的編號。

輸出

首先在第一行輸出家庭個數（所有有親屬關係的人都屬於同一個家庭）。隨後按下列格式輸出每個家庭的資訊：

家庭成員的最小編號 家庭人口數 人均房產套數 人均房產面積

其中人均值要求保留小數點後3位。家庭資訊首先按人均面積降序輸出，若有並列，則按成員編號的升序輸出。

樣例輸入

10
6666 5551 5552 1 7777 1 100
1234 5678 9012 1 0002 2 300
8888 -1 -1 0 1 1000
2468 0001 0004 1 2222 1 500
7777 6666 -1 0 2 300
3721 -1 -1 1 2333 2 150
9012 -1 -1 3 1236 1235 1234 1 100
1235 5678 9012 0 1 50
2222 1236 2468 2 6661 6662 1 300
2333 -1 3721 3 6661 6662 6663 1 100

樣例輸出

3
8888 1 1.000 1000.000
0001 15 0.600 100.000
5551 4 0.750 100.000

分析

DFS，注意要懂得將人與人的關係和人的資訊分離出來，分開儲存，vector<int>g[N]，存放所有人的關係，比如a是b的父親，則g[a]中插入b，g[b]中插入a，這樣就能記錄兩者的關係，即在a與b之間有一條雙向邊。存放人資訊的結點只需要記錄這個人擁有的房子套數和總面積，人的編號用陣列的下標記錄即可

用vis[i]=true記錄編號為i的人出現過，題中可能有這樣的情況：某個人只會以某個結點的父親或母親結點出現，而不會單獨給出一條描述此結點的資訊，因此我們需要將這些結點也記錄下來，預設其擁有的房產套數和房產面積為0

處理好後，我們就能進行DFS了，每次只要遇到一個vis[i]=true的結點，就從結點i進行DFS，將它遍歷到的點的vis值置為false，並統計相關的資訊，壓入存放結果的向量即可。

C++程式

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

const int N=10009;

//a[i]存放編號為i的人擁有的房子套數以及房子總面積 
struct Node{
	int num;
	double area;
}a[N];

//存放結果 
struct Res{
	int min_id,people;
	double ave_house,ave_area;
	Res(int min_id,int people,double ave_house,double ave_area)
	{
		this->min_id=min_id;
		this->people=people;
		this->ave_house=ave_house;
		this->ave_area=ave_area;
	}
	//家庭資訊首先按人均面積降序排列，若有並列，則按成員編號的升序排列 
	bool operator <(const Res &h)const
	{
		return ave_area==h.ave_area?(min_id<h.min_id):(ave_area>h.ave_area);
	}
};

vector<int>g[N];//使用鄰接表記錄關係 
vector<Res>v; //記錄結果 
bool vis[N];//vis[i]=true表示編號為i的人出現過 

int min_id,people;
double ave_area,ave_house;

void dfs(int u)
{
	vis[u]=false;//標記結點u已經被訪問了 
	
	//統計此結點的資訊 
	if(min_id>u) min_id=u;//更新最小編號 
	people+=1;//家庭人口數加一 
	ave_house+=a[u].num;//加上房子套數 
	ave_area+=a[u].area;//加上總面積 
	
	for(int i=0;i<g[u].size();i++)//統計u的父母親和孩子結點 
	{
		int v=g[u][i];
		if(vis[v]) dfs(v);
	}
}

int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		int x,f,m,k,id;
		scanf("%d%d%d%d",&x,&f,&m,&k);
		vis[x]=true;//標記結點x出現過 
		if(f!=-1) g[f].push_back(x),g[x].push_back(f),vis[f]=true;
		if(m!=-1) g[m].push_back(x),g[x].push_back(m),vis[m]=true;
		while(k--)
		{
			scanf("%d",&id);
			vis[id]=true; 
			g[x].push_back(id);
			g[id].push_back(x);
		}
		scanf("%d%lf",&a[x].num,&a[x].area);
	}
	for(int i=0;i<N;i++)
	  if(vis[i])
	  {
	  	min_id=10000,people=0,ave_house=0,ave_area=0;
	  	dfs(i);
	  	ave_area/=people,ave_house/=people;
	  	v.push_back(Res(min_id,people,ave_house,ave_area));
	  }
	sort(v.begin(),v.end());
	printf("%d\n",v.size()); 
	for(int i=0;i<v.size();i++)
	  printf("%04d %d %.3lf %.3lf\n",v[i].min_id,v[i].people,v[i].ave_house,v[i].ave_area);
	return 0;
}