【PAT】1114. Family Property

考查点：并查集

思路及提交情况：这道题直接用并查集即可，但是在合并的时候要保证指向ｉｄ最大的节点，还有合并同时必须完成房产个数和房地的更新，这里还用到了ｍａｐ把ｉｄ映射为下标便于遍历，但是由于多了个映射第一次提交时候忘记把比较根节点时候先映射成ｉｄ号，第二次提交时候房地产没有更新，原因也是在获取ｒｏｏｔ值时没有用ｍｐ把ｉｄ映射成下标号，最后提交有一组错误。以为是边界数据，实际上是程序写错了，以后要过一遍代码，看有没有错别字，这里就是把数组名写错结果还有２３分。。

#define LOCAL
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <stack>
#define FOR(i, x, y) for(int i = x; i <= y; i++)
#define rFOR(i, x, y) for(int i = x; i >= y; i--)
#define MAXN 10010
#define oo 0x3f3f3f3f
using namespace std;


int fa[MAXN];
int findRoot(int a)
{
    int z=a;
    while(a!=fa[a])a=fa[a];
    while(z!=fa[z])
    {
        int x=z;
        z=fa[z];
        fa[x]=a;
    }
    return a;
}
map<int,int> mp,pm;
double est[MAXN];
double marea[MAXN];
void Union(int a,int b)
{
    int ra=findRoot(a),rb=findRoot(b);
    if(ra!=rb){
        if(pm[ra]<pm[rb]){
            fa[rb]=ra;
            est[ra]+=est[rb];
            marea[ra]+=marea[rb];
        }else{
            fa[ra]=rb;
            est[rb]+=est[ra];
            marea[rb]+=marea[ra];
        }
    }
}
void init()
{
    FOR(i,0,MAXN)
    {
        fa[i]=i;
    }
}
int isRoot[MAXN];
int record[MAXN];
bool cmp(int a,int b)
{
    if(marea[a]!=marea[b])return marea[a]>marea[b];
    else if(pm[a]!=pm[b]) return pm[a]<pm[b];
}

int main()
{
     #ifdef LOCAL
        freopen("data.in","r",stdin);
        freopen("data.out","w",stdout);
    #endif // LOCAL
    int n;
    init();
    int id,fr,mr,k,ch;
    double area;double m;
    scanf("%d",&n);
    int num=0;
    FOR(i,1,n)
    {

        scanf("%d%d%d%d",&id,&fr,&mr,&k);
        if(mp.find(id)==mp.end()){
            mp[id]=num;
            pm[num]=id;
            num++;
        }
        if(fr!=-1){
            if(mp.find(fr)==mp.end()){
                mp[fr]=num;
                pm[num]=fr;
                num++;
            }
            Union(mp[id],mp[fr]);

        }
        if(mr!=-1){

            if(mp.find(mr)==mp.end()){
                mp[mr]=num;
                pm[num]=mr;
                num++;
            }
            Union(mp[id],mp[mr]);
        }

        FOR(i,1,k)
        {
            scanf("%d",&ch);
            if(mp.find(ch)==mp.end()){
                mp[ch]=num;
                pm[num]=ch;
                num++;
            }
            Union(mp[ch],mp[id]);
        }
        scanf("%lf%lf",&m,&area);
        int root=findRoot(mp[id]);
        est[root]+=m;
        marea[root]+=area;

    }

    int cnt=0;
    FOR(i,0,num-1)
    {
        isRoot[findRoot(i)]++;
    }

    FOR(i,0,num-1)
    {
        if(isRoot[i]){
            record[cnt++]=i;
            est[i]/=isRoot[i];
            marea[i]/=isRoot[i];

        }
    }

    sort(record,record+cnt,cmp);
    printf("%d\n",cnt);
    FOR(i,1,cnt)
    {
        int id=record[i-1];
        printf("%04d %d %.3f %.3f\n",pm[id],isRoot[id],est[id],marea[id]);
    }
    return 0;
}