C语言：L2-007 家庭房产 (25 分) — 并查集

一、题目

定每个人的家庭成员和其自己名下的房产，请你统计出每个家庭的人口数、人均房产面积及房产套数。

输入格式：
输入第一行给出一个正整数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

二、方法1

1、思路

（1）整体架构

在输入时使用并查集的思维，将数据存放至其中，找到每个家庭最小的成员，用该成员代表整个家庭，先算房产总和，同时计算家庭人数，再计算个人房产，然后创建一个结构体 NODE2，存放将 NODE1 中无效数据清除以后的数据，然后按照题目要求排序即可。

（2）初始化

由于需要使用 NODE1 中的 flag 来判断该家庭是否遍历，，以及并查集用到的数组 home 都需要初始化，所以单独写一个初始化函数。

具体实现：

void init() {
	for (int i = 0; i < 10010; i++)
	{
		home[i] = i;
		b[i].flag = 0;
	}
}

（3）类并查集思维

前提：

int find(int x);
void Union(int a, int b);
int home[10010]; 

具体实现：

int find(int x)
{
	while (x != home[x])
		x = home[x];
	return x;
}

void Union(int a, int b)
{
	int f_a = find(a);
	int f_b = find(b);
	if (f_a > f_b)
		home[f_a] = f_b;
	else if (f_a < f_b)
		home[f_b] = f_a;
}

（4）排序

这里其实可以直接用系统给的 qsort 函数，博主尝试了很久都没有成功，所以直接手动实现了。

具体实现：

NODE2 t;
for (int i = 0; i < count; i++)
{
	for (int j = 0; j < count; j++)
	{
		if (c[j].area < c[j + 1].area)
		{
			t = c[j];
			c[j] = c[j + 1];
			c[j + 1] = t;
		}
		else if (c[j].area == c[j + 1].area)
		{				
			if (c[j].id > c[j + 1].id)
			{
				t = c[j];
				c[j] = c[j + 1];
				c[j + 1] = t;
			}
		}
	}
}

2、代码

#include<stdio.h>
#include<stdlib.h>

 快排
//int cmp(const void* e1, const void* e2);

typedef struct node
{
	int people;
	int fa;
	int ma;
	int kid_number;
	double sum, area;
	int kid[10];
} NODE;
NODE a[10010];

typedef struct node1
{
	double area, sum;
	int person, id, flag;
} NODE1;
NODE1 b[10010];

typedef struct node2
{
	double area, sum;
	int person, id;
} NODE2;
NODE2 c[110];

int N;
int i, j, k;				// 遍历
int count = 0, cnt = 0; 	// 家庭个数
int visit[10010] = { 0 };   // 某人是否存在
int home[10010];            // 并查集所用数组

// 并查集
void init();
int find(int x);
void Union(int a, int b);

int main()
{
	// 初始化
	init();

	scanf("%d", &N);
	for (i = 0; i < N; i++)
	{
		scanf("%d %d %d %d", &a[i].people, &a[i].fa, &a[i].ma, &a[i].kid_number);
		visit[a[i].people] = 1;

		// 有爸爸
		if (a[i].fa != -1)
		{
			Union(a[i].people, a[i].fa);
			visit[a[i].fa] = 1;
		}

		// 有妈妈
		if (a[i].ma != -1)
		{
			Union(a[i].people, a[i].ma);
			visit[a[i].ma] = 1;
		}

		// 孩子
		for (j = 0; j < a[i].kid_number; j++)
		{
			scanf("%d", &a[i].kid[j]);
			Union(a[i].people, a[i].kid[j]);
			visit[a[i].kid[j]] = 1;
		}

		// 房产套数 和 总面积
		scanf("%lf %lf", &a[i].sum, &a[i].area);
	}

	// 以孩子为下标计算全家房产
	for (int i = 0; i < N; i++)
	{
		int id = find(a[i].people);
		b[id].id = id;
		b[id].sum += a[i].sum;
		b[id].area += a[i].area;
		b[id].flag = 1;
	}

	// 计算家庭人数
	for (i = 0; i < 10010; i++)
	{
		if (visit[i] != 0)
			b[find(i)].person++;
		if (b[i].flag != 0)
			cnt++;
	}

	// 计算个人资产
	for (i = 0; i < 10010; i++) {
		if (b[i].flag)
		{
			b[i].sum = b[i].sum * 1.0 / b[i].person;
			b[i].area = b[i].area * 1.0 / b[i].person;
		}
	}

	// 新建结构体，留下有效项
	for (int i = 0; i < 10010; i++)
	{
		if (b[i].flag != 0)
		{
			c[count].id = b[i].id;
			c[count].person = b[i].person;
			c[count].sum = b[i].sum;
			c[count].area = b[i].area;
			count++;
		}
		else
			continue;
	}

	//qsort(c, max, sizeof(c[0]), cmp);

	NODE2 t;
	for (int i = 0; i < count; i++)
	{
		for (int j = 0; j < count; j++)
		{
			if (c[j].area < c[j + 1].area)
			{
				t = c[j];
				c[j] = c[j + 1];
				c[j + 1] = t;
			}
			else if (c[j].area == c[j + 1].area)
			{
				if (c[j].id > c[j + 1].id)
				{
					t = c[j];
					c[j] = c[j + 1];
					c[j + 1] = t;
				}
			}
		}
	}
	printf("%d\n", count);
	for (int i = 0; i < count; i++)
	{
		printf("%04d %d %.3lf %.3lf\n", c[i].id, c[i].person, c[i].sum, c[i].area);
	}
	return 0;
}
void init() {
	for (int i = 0; i < 10010; i++)
	{
		home[i] = i;
		b[i].flag = 0;
	}
}

int find(int x)
{
	while (x != home[x])
		x = home[x];
	return x;
}

void Union(int a, int b)
{
	int f_a = find(a);
	int f_b = find(b);
	if (f_a > f_b)
		home[f_a] = f_b;
	else if (f_a < f_b)
		home[f_b] = f_a;
}

//int cmp(const void* e1, const void* e2)
//{
//	node3* a = (node3*)e1;
//	node3* b = (node3*)e2;
//	if (a->house_area != b->house_area)
//		return ((a->house_area) - (b->house_area));
//	else
//		return ((b->people) - (a->people));
//}