*浙大PAT甲级 1114 并查集

考察并查集,并附加记录额外信息。

为了得到一个家族中的最小id,我们可以将最小的id做为这个家族的根。并将整个家族的信息如家族人数,家族房产数,家族的房产面积。

然后通过排序,输出相应结果。

AC代码:

#include<iostream>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#include<hash_map>
#define ll long long
#define inf 24*60*60
using namespace std;
int father[10000];
int mark[10000];
struct node
{
    int id;
    int shu=1;
    double sets=0;
    double area=0;
};
node bb[10000];
vector<node> v;
int findroot(int x)
{
    if(father[x]==-1) return x;
    else
    {
        int tmp=findroot(father[x]);
        father[x]=tmp;
        return tmp;
    }
}
bool cmp(node x,node y)
{
    if(fabs(x.area/x.shu-y.area/y.shu)<1e-5)
    {
        return x.id<y.id;
    }
    else
    {
        return x.area/x.shu>y.area/y.shu;
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    memset(father,-1,sizeof(father));
    for(int i=0;i<n;i++)
    {
        int d;
        scanf("%d",&d);
        int lp,rp;
        scanf("%d %d",&lp,&rp);
        int root1=findroot(d);
        bb[d].id=d;
        mark[d]=1;
        if(lp!=-1)
        {
            int root2=findroot(lp);
            mark[lp]=1;
            if(root1!=root2)
            {
            int minn=min(root1,root2);
            int maxx=max(root1,root2);
            father[maxx]=minn;
            bb[minn].sets+=bb[maxx].sets;
            bb[maxx].sets=0;
            bb[minn].area+=bb[maxx].area;
            bb[maxx].area=0;
            bb[minn].shu+=bb[maxx].shu;
            bb[maxx].shu=0;
            bb[minn].id=minn;
            root1=minn;
            }
        }
        if(rp!=-1)
        {
            int root2=findroot(rp);
            mark[rp]=1;
            if(root1!=root2)
            {
            int minn=min(root1,root2);
            int maxx=max(root1,root2);
            father[maxx]=minn;
            bb[minn].sets+=bb[maxx].sets;
            bb[maxx].sets=0;
            bb[minn].area+=bb[maxx].area;
            bb[maxx].area=0;
            bb[minn].shu+=bb[maxx].shu;
            bb[maxx].shu=0;
            bb[minn].id=minn;
            root1=minn;
            }
        }
        int k;
        scanf("%d",&k);
        for(int j=0;j<k;j++)
        {
            int c;
            scanf("%d",&c);
            mark[c]=1;
            int root2=findroot(c);
            if(root1!=root2)
            {
            int minn=min(root1,root2);
            int maxx=max(root1,root2);
            father[maxx]=minn;
            bb[minn].sets+=bb[maxx].sets;
            bb[maxx].sets=0;
            bb[minn].area+=bb[maxx].area;
            bb[maxx].area=0;
            bb[minn].shu+=bb[maxx].shu;
            bb[maxx].shu=0;
            bb[minn].id=minn;
            root1=minn;
            }
        }
        double sets;
        double area;
        scanf("%lf %lf",&sets,&area);
        bb[root1].sets+=sets;
        bb[root1].area+=area;
        //cout<<root1<<endl;
    }
    //cout<<1<<endl;
    int ans=0;
    for(int i=0;i<10000;i++)
    {
        if(mark[i]==1&&bb[i].shu!=0)
        {
            ans++;
            v.push_back(bb[i]);
        }
    }
    sort(v.begin(),v.end(),cmp);
    printf("%d\n",ans);
    for(int i=0;i<v.size();i++)
    {
        printf("%04d %d %.3lf %.3lf\n",v[i].id,v[i].shu,v[i].sets/v[i].shu,v[i].area/v[i].shu);
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值