PAT---甲级1034 Head of a Gang 30分 并查集使用练习题

题目信息:
  One way that the police finds the head of a gang is to check people’s phone calls. If there is a phone call between A and B, we say that A and B is related. The weight of a relation is defined to be the total time length of all the phone calls made between the two persons. A “Gang” is a cluster of more than 2 persons who are related to each other with total relation weight being greater than a given threshold K. In each gang, the one with maximum total weight is the head. Now given a list of phone calls, you are supposed to find the gangs and the heads.

  Input Specification:

  Each input file contains one test case. For each case, the first line contains two positive numbers N and K (both less than or equal to 1000), the number of phone calls and the weight threshold, respectively. Then N lines follow, each in the following format:

  Name1 Name2 Time

  where Name1 and Name2 are the names of people at the two ends of the call, and Time is the length of the call. A name is a string of three capital letters chosen from A-Z. A time length is a positive integer which is no more than 1000 minutes.

  Output Specification:

  For each test case, first print in a line the total number of gangs. Then for each gang, print in a line the name of the head and the total number of the members. It is guaranteed that the head is unique for each gang. The output must be sorted according to the alphabetical order of the names of the heads.

  Sample Input 1:

8 59
AAA BBB 10
BBB AAA 20
AAA CCC 40
DDD EEE 5
EEE DDD 70
FFF GGG 30
GGG HHH 20
HHH FFF 10

  Sample Output 1:

2
AAA 3
GGG 3

  题目大意:警察找犯罪头目的一种方式是检查人们的通话,如果两个人可以应为通话关系直接或间接关联,那么这两个人就在同一个团队里。如果一个团队的人数大于2并且总通话量大于给定值,那么这个团队就是一个gang,并且gang里面单人通话量最多的人是head,题目会给出通话关系,要求找出所有的head
输入格式:先输入两个int型,分别表示即将给出的通话关系数量和满足gang的通话量阈值,然后给出所有的通话关系,以 人名甲 人名乙 权值 的格式给出。
输出格式:先输出gang的数量,然后输出每组gang的head和该gang的人数。

这里写图片描述

  如图,当画出了所给示例的图后,我们发现每个团体就是一个连通块,我们要做的就是遍历连通块获取相关信息。本题可以用DFS搜索图的解法做,不过在这里我主要介绍下并查集解决此类问题的解法。

   首先我们需要一个数组作为并查集,下标表示人,(这里需要一个从string的名字到int型的转换,可以用hash映射好处是方便后面字典排序但是不方便反转,或者用STL的map做)数组值表示这个人所在的gang的head,也就是说一个连通块的所有成员的数组值都是head,这是并查集的quick-find类型(详见算法第四版)。
   为了找到每个连通块的其他信息,可以在创建好并查集之后再从并查集中分别寻找每个连通块的成员再从中找到人数和总通话量信息。不过有个更好的一步到位的方法,就是再设置两个数组ufs_num和ufs_total,分别表示成员i所在的团体的人员数量和团体总通话量,这样我们只用在处理好输入数据的同时对数组元素进行修改,之后就得到了所有信息。
   我们可以看出最关键的步骤就是在处理输入的同时完成并查集数组并更新ufs_num和ufs_total,但是这里稍有不同,我们要对输入数据进行两次分析,第一次完成对人员名字从string到int的转换,和个人点权的更新,以及把边存起来,第二次才是对并查集和两个数组的更新。原因是我们要保证并查集每个人指向自己团队的head,只有先知道了所有人的点权才能这样做。
   要注意的是在做并查集union操作的时候逻辑有些多,如果两个点是在同一个连通块里,要更新ufs_total数组不更新ufs_num数组,然后退出,如果不在同一个连通块里,更新并查集,更新ufs_num和ufs_total

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <map>
#include <string> 
#include <vector>
using namespace std;
//结构体edge保存输入信息
struct edge{
    int from;
    int to;
    int value;
};

const int maxn=2010;            //题目说输入边的数量最多1000,最坏情况是每次都不相同那么就是2000人
int ufs[maxn],weight[maxn];     //ufs并查集数组,weight数组存储每个人的点权
int n,kk;                       //n表示输入的通话数,kk表示团体满足一个gang的阈值
int countt=0,edgecount=0,gangcount=0;   //分别表示人员总数,通话总数,满足gang团体总数
vector<edge> alledge;           //存储所有通话边的向量
int ufs_num[maxn]={1};          //表示人员i所在的团体的人员数
int ufs_total[maxn]={0};        //表示人员i所在的团体的通话总量
map<string,int> STI;            //将题目输入的人名转换成int
map<int,string> ITS;            //将代表人的int和名字的键值对

//最后要把所有head按字典序输出,使用algorithm头文件的sort函数
bool cmp(int a,int b)       
{
    string sa=ITS[a];
    string sb=ITS[b];
    for(int i=0;i<3;i++)
    {
        char ca=sa[i];
        char cb=sb[i];
        if(ca!=cb) return ca<cb;
    }
}

//并查集查找 因为是quick-find所以直接一步查找到位
int find(int k)
{
    return ufs[k];
}
//并查集合并操作,在本题里不止做合并,还有ufs_total和ufs_num数组的更新,传入的参数分别是通话记录里两个人和本次通话权值
void unionn(int a,int b,int v)
{
    int ra=find(a);
    int rb=find(b);
    if(ra==rb)               //如果两个人在同一个连通块里,那么找到该连通块所有的人,把ufs_total加上这次边的权值
    {
        for(int i=0;i<maxn;i++)
        {
            if(ufs[i]==ra) ufs_total[i]+=v;
        } 
        return;
    }
    int oldat=ufs_total[ra],oldbt=ufs_total[rb];
    int oldan=ufs_num[ra],oldbn=ufs_num[rb];
    if(weight[ra]>=weight[rb])
    {
        for(int i=0;i<maxn;i++)
        {
            if(ufs[i]==rb) {
                ufs[i]=ra;ufs_total[i]+=oldat+v;ufs_num[i]+=oldan;
            }
            else if(ufs[i]==ra){
                ufs_total[i]+=oldbt+v;ufs_num[i]+=oldbn;
            } 
        }
    } else if(weight[ra]<weight[rb]){
        for(int i=0;i<maxn;i++)
        {
            if(ufs[i]==ra) 
            {
                ufs[i]=rb;ufs_total[i]+=oldbt+v;ufs_num[i]+=oldbn;
            }
            else if(ufs[i]==rb) 
            {
                ufs_total[i]+=oldat+v;ufs_num[i]+=oldan;
            }

        }
    }
}
//在第一次读入输入数据时调用,把人名转换成int
int nametoint(string s)
{
    if(STI.find(s)!=STI.end()) return STI[s];
    else {
        STI[s]=countt;
        ITS[countt]=s;
        return countt++;
    }
}

int main()
{
    //初始化并查集,每个人指向自己
    for(int i=0;i<maxn;i++)
    {
        ufs[i]=i;
    }
    scanf("%d %d",&n,&kk);
    char a[4],b[4];
    string sa,sb;
    int ia,ib,value;
    for(int i=0;i<n;i++)  //第一次处理输入数据。至于为什么要用char数组转一次string。。因为不喜欢用cin
    {
        scanf("%s %s %d",a,b,&value);
        sa=a;sb=b;
        //把string的名字转换成int
        ia=nametoint(sa);
        ib=nametoint(sb);
        //把通话信息整合成结构体存在向量里
        edge e;edgecount++;
        e.from=ia;e.to=ib;e.value=value;
        alledge.push_back(e);
        //更新点权
        weight[ia]+=value;
        weight[ib]+=value;
    }

    fill(ufs_num,ufs_num+maxn,1);
    //第二次处理输入数据,做并查集合并
    for(int i=0;i<alledge.size();i++)
    {
        ia=alledge[i].from;
        ib=alledge[i].to;
        value=alledge[i].value;
        unionn(ia,ib,value);
    }
    int visit[maxn]={0}; //用数组简单做判断:这个团体的信息是否已经处理过
    int gangi[maxn];     //当找到一个满足条件的gang,把它的head加入到这个数组,最后再做输出,因为要按字典序,不能找到一个就直接输出
    int gangsize=0;
    for(int i=0;i<maxn;i++)
    {
        if(visit[ufs[i]]==0)
        {
            visit[ufs[i]]=1;
            if(ufs_num[ufs[i]]>2)
            {
                if(ufs_total[ufs[i]]>kk)
                {
                    gangi[gangsize++]=ufs[i]; //遍历并查集如果当前点指向的head没被处理过,就进入检查人员和通话总量是否满足条件
                }
            }
        }
    }
    sort(gangi,gangi+gangsize,cmp);//按字典序排序
    printf("%d\n",gangsize);
    for(int i=0;i<gangsize;i++)
    {
        cout<<ITS[gangi[i]]<<" "<<ufs_num[gangi[i]]<<endl;
    }
}

这里写图片描述

展开阅读全文

没有更多推荐了,返回首页