天梯赛 并查集相关题目

L2-007 家庭房产 （25 分)

 

 

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

输入格式：

输入第一行给出一个正整数N（≤1000），随后N行，每行按下列格式给出一个人的房产：

其中编号是每个人独有的一个4位数的编号；父和母分别是该编号对应的这个人的父母的编号（如果已经过世，则显示-1）；k（0≤k≤5）是该人的子女的个数；孩子i是其子女的编号。

输出格式：

首先在第一行输出家庭个数（所有有亲属关系的人都属于同一个家庭）。随后按下列格式输出每个家庭的信息：

其中人均值要求保留小数点后3位。家庭信息首先按人均面积降序输出，若有并列，则按成员编号的升序输出。

 

恶心。。。

用并查集合并当前这个人和他的父母以及儿子，合并为最小的下标

然后再开一个数组去搞会方便一点

#include <iostream>
#include <queue>
#include <map>
#include <cmath>
#include <set>
#include <iomanip>
#include <iomanip>
#include <unordered_set>
#include <cstring>
#include <unordered_map>
#include <algorithm>
using namespace std;
const int maxn=10005;
int n;
int pre[maxn];

struct node
{
    int id;
    int fa=0,ma=0;
    int num=0;
    double area=0;
    int peo=0;
    double areas=0;
}a[maxn],ans[maxn];

int uni(int root)
{
    int son,tem;
    son=root;
    while(root!=pre[root])
    {
        root=pre[root];
    }
    while(son!=root)
    {
        tem=pre[son];
        pre[son]=root;
        son=tem;
    }
    return root;
}

void join(int x,int y)
{
    x=uni(x);
    y=uni(y);
    if(x>y)
    {
        pre[x]=y;
    }
    else  pre[y]=x;
}

bool cmp(node a,node b)
{
    if(a.areas==b.areas)
    {
        return a.id<b.id;
    }
    return a.areas>b.areas;
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    int k;
    int summ=0;
    for(int i=0;i<10000;i++)pre[i]=i;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i].id>>a[i].fa>>a[i].ma>>k;
        if(a[i].fa!=-1)
        {
            join(a[i].fa,a[i].id);
        }
        if(a[i].ma!=-1)
        {
            join(a[i].ma,a[i].id);
        }
        int son;
        for(int j=1;j<=k;j++)
        {
            cin>>son;
            join(a[i].id,son);
        }
        cin>>a[i].num>>a[i].area;
    }
    int x;
    for(int i=1;i<=n;i++)
    {
        x=uni(a[i].id);//找根节点
        ans[x].id=x;
        ans[x].area+=a[i].area;
        ans[x].num+=a[i].num;
        ans[x].fa=-1;
    }
    //cout<<"summ"<<summ<<endl;
    for(int i=0;i<10000;i++)
    {
        ans[uni(i)].peo++;
        if(ans[i].fa==-1){
            summ++;
        }
    }
    for(int i=0;i<10000;i++)
    {
        if(ans[i].fa==-1)
        {
            ans[i].areas=ans[i].area/ans[i].peo;
        }
    }
    cout<<summ<<endl;
    sort(ans,ans+9999,cmp);
    for(int i=0;i<summ;i++)
    {
        cout<<setfill('0')<<setw(4)<<ans[i].id;
        cout<<' '<<ans[i].peo<<' ';
        cout<<fixed<<setprecision(3)<<(double)ans[i].num/ans[i].peo<<' '
        <<ans[i].areas<<endl;
    }



    return 0;
}

 

L2-010 排座位 （25 分)

布置宴席最微妙的事情，就是给前来参宴的各位宾客安排座位。无论如何，总不能把两个死对头排到同一张宴会桌旁！这个艰巨任务现在就交给你，对任何一对客人，请编写程序告诉主人他们是否能被安排同席。

输入格式：

输入第一行给出3个正整数：N（≤100），即前来参宴的宾客总人数，则这些人从1到N编号；M为已知两两宾客之间的关系数；K为查询的条数。随后M行，每行给出一对宾客之间的关系，格式为：宾客1 宾客2 关系，其中关系为1表示是朋友，-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K行，每行给出一对需要查询的宾客编号。

这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友，朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。

输出格式：

对每个查询输出一行结果：如果两位宾客之间是朋友，且没有敌对关系，则输出No problem；如果他们之间并不是朋友，但也不敌对，则输出OK；如果他们之间有敌对，然而也有共同的朋友，则输出OK but...；如果他们之间只有敌对关系，则输出No way。

 

 

并查集朋友，因为朋友的朋友也是朋友，对于敌人直接标记

 

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e3+5;
int pre[maxn];
set<int>s[maxn];
int uni(int root)
{
    int son, tem;

        son = root;
        while (root != pre[root])
        {
            root = pre[root];
        }
        while (son != root)
        {
            tem = pre[son];
            pre[son] = root;
            son = tem;
        }
        return root;
}

int main()
{
    ios::sync_with_stdio(false);
    int n,m,k;
    int x,y,tem;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    {
        pre[i]=i;
    }
    for(int i=1;i<=m;i++)
    {
        int xx,yy;
        cin>>x>>y>>tem;
        if(tem==1)
        {
            xx=uni(x);
            yy=uni(y);
            pre[xx]=yy;
        }
        else
        {
            s[y].insert(x);
            s[x].insert(y);
        }
    }
    for(int i=1;i<=k;i++)
    {
        cin>>x>>y;
        if(!s[x].empty()&&s[x].find(y)!=s[x].end())
        {
            if(uni(x)!=uni(y))
            {
                cout<<"No way"<<endl;
            }
            else cout<<"OK but..."<<endl;
        }
        else
        {
            if(uni(x)==uni(y))
            {
                cout<<"No problem"<<endl;
            }
            else cout<<"OK"<<endl;
        }
    }


    return 0;
}

 

L2-013 红色警戒

并查集一下记录个数，如果删除前的个数=删除后的个数或者删除前的个数=删除后的个数-1，那么删除这个点就没有影响

其他情况是有影响的

 

#include <bits/stdc++.h>
#define quickio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
#define ll long long
int n,m;
const  int maxn=5005;
int pre[maxn];

int uni(int x)
{
    return x==pre[x]?pre[x]:uni(pre[x]);
}
void join(int x,int y)
{
    int xx=uni(x);
    int yy=uni(y);
    pre[xx]=yy;
}

int main()
{
    int u[maxn],v[maxn];
    cin>>n>>m;
    for(int i=0;i<n;i++)pre[i]=i;

    for(int i=1;i<=m;i++)
    {
        cin>>u[i]>>v[i];
        join(u[i],v[i]);
    }

    int k;
    int a;
    int num=0;
    int new_num=0;

    cin>>k;
    for(int i=0;i<n;i++)
    {
        if(pre[i]==i)
        {
            num++;
        }
    }
    set<int>s;

    for(int i=1;i<=k;i++)
    {
        new_num=0;
        cin>>a;
        s.insert(a);
        for(int j=0;j<n;j++)pre[j]=j;

        for(int j=1;j<=m;j++)
        {
            if(s.find(u[j])!=s.end()||s.find(v[j])!=s.end())
                continue;
            join(u[j],v[j]);
        }
        for(int j=0;j<n;j++)
        {
            if(s.find(j)!=s.end())continue;
            if(pre[j]==j)
            {
                new_num++;
            }
        }

        if(new_num==num||new_num==num-1)
        {
            cout<<"City "<<a<<" is lost."<<endl;
        }
        else cout<<"Red Alert: City "<<a<<" is lost!"<<endl;
        if((int)s.size()==n)
        {
            cout<<"Game Over."<<endl;
        }
        num=new_num;
    }


    return 0;
}