2019新生集训第一场 7-11 家庭房产（有问题）

我写的答案有问题，开始时用set容器装最终所有的户主，测试点3段错误。我又把set改为数组，结果测试点1答案错误。
这道题浪费太多时间啦，先把题放在这吧，过阵子再琢磨琢磨。

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

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

//并查集
#include <bits/stdc++.h>
using namespace std;
int min_id[11000];//家庭成员最小编号
int number[11000];//家庭人口数
int house_num[11000];//各个家庭房产总套数
int house_s[11000];//房产总面积
int root[11000];//户主
int is_root[11000];
int a[11000];//存所有户主的id
int k = 0;
int s[500000];
int ss = 0;
int find_fa(int x)//找x的户主
{
    int r = x;
    while(root[r]!=r)
        r = root[r];
    //此时的r为户主
    while(x!=r)
    {
        int tmp = root[x];
        root[x] = r;//户主改为真正的户主
        x = tmp;
    }
    is_root[r] = 1;//标记r为户主
    return r;
}
void link(int q1,int q2)
{
    int x = find_fa(q1);
    int y = find_fa(q2);
    if(x==y)
        return;
    if(x>y)
    {
        root[x] = y;
        is_root[x] = 0;//x不再是户主
        number[y] += number[x];
        house_num[y]+=house_num[x];
        house_s[y]+=house_s[x];
    }
    else
    {
        root[y] = x;
        is_root[y] = 0;//x不再是户主
        number[x] += number[y];
        house_num[x]+=house_num[y];
        house_s[x]+=house_s[y];
    }
}
bool cmp(int x,int y)
{
    //人均房产面积
    double ave_x = 1.0*house_s[x]/number[x];
    double ave_y = 1.0*house_s[y]/number[y];
    if(ave_x==ave_y)
        return x<y;
    else return x>y;
}
void print()
{
    for(int i = 0; i<k; i++)
    {
        int r = a[i];
        printf("%04d %d ",r,number[r]);
        double n1 = 1.0*house_num[r]/number[r];//人均房产套数
        double ave_x = 1.0*house_s[r]/number[r];//人均房产面积
        printf("%.3lf %.3lf\n",n1,ave_x);
    }
}
int main()
{
    int N;
    cin>>N;
    memset(house_num,0,sizeof(house_num));
    memset(house_s,0,sizeof(house_s));
    for(int i = 0; i<=9999; i++)
    {
        root[i] = i;
        min_id[i] = i;
        number[i] = 1;
    }
    for(int i = 0; i<N; i++)
    {
        int a,b,c,k;
        cin>>a>>b>>c>>k;
        s[ss++] = a;
        if(b!=-1)
        {
            s[ss++] = b;
            link(a,b);
        }
        if(c!=-1)
        {
            s[ss++] = c;
            link(a,c);
        }
        int mm;
        for(int j = 0; j<k; j++)
        {
            cin>>mm;
            s[ss++] = mm;
            link(a,mm);
        }
        int av1,av2;
        cin>>av1>>av2;
        int fa = find_fa(a);
        house_num[fa]+=av1;
        house_s[fa]+=av2;
    }
    //遍历找所有户主
    for(int i = 0;i<ss;i++)
    {
        if(is_root[s[i]]==1)
            a[k++] = s[i];
    }
    //for(int i = 0;i<k;i++)
        //cout<<a[i]<<endl;
    sort(a,a+k,cmp);
    //for(int i = 0;i<k;i++)
        //cout<<a[i]<<endl;
    cout<<k<<endl;
    print();
}