1034 Head of a Gang(并查集)

最新推荐文章于 2024-09-14 22:25:23 发布

饼饼夹豆腐

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量70

点赞数

分类专栏： PAT 文章标签：算法数据结构

本文链接：https://blog.csdn.net/cxw789/article/details/120079131

版权

PAT 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

1034 Head of a Gang

题意描述

由通话信息来判断几人是否位于一个帮派

输入输出格式

输入通话记录输出帮派的个数以及每个帮派的头目和人数

数据规模

算法设计

这里采用并查集的做法。首先建立并查集数组unordered_map<string,string>ufs 再维护每个人的权重 unordered_map<string,gg>weight 最后由于题目要求按头目字典序升序输出故用 map<string,array<gg,2>>res 维护结果头目对应人数 + 帮派的权值值得注意的是不能边读入边进行union 因为此时的结点权重信息是不完整的而且帮派的权重一定为帮派结点权重 / 2

c++11代码

#include <bits/stdc++.h>
using namespace std;
using gg = long long;
unordered_map<string,string>ufs;   //   并查集数组
unordered_map<string,gg>weight;    //   结点的权重
map<string,array<gg,2>>res;        //   结果 father 对应 人数 + 权重之和
string findFather(string x){
    return x == ufs[x] ? x : ufs[x] = findFather(ufs[x]);
}
void unionSets(string x, string y){
    auto a = findFather(x), b = findFather(y);
    if(a != b) weight[a] > weight[b] ? ufs[b] = a : ufs[a] = b;  
}
int main()
{
   ios::sync_with_stdio(false);
   cin.tie(0);
   gg n,k,time;
   string name1,name2;
   vector<array<string,2>>v;  
   cin>>n>>k;
   while(n--){
       cin>>name1>>name2>>time;
       if(not ufs.count(name1)) ufs[name1] = name1;
       if(not ufs.count(name2)) ufs[name2] = name2;
       weight[name1] += time;
       weight[name2] += time;
       v.push_back({name1,name2});   //   这里要注意 由于结点的权重不完整 不能在此处进行unionSets 
   }
   for(auto& i : v) unionSets(i[0],i[1]);
   for(auto& i : ufs){  //  统计 father 信息
       res[findFather(i.first)][1] += weight[i.first];
       res[findFather(i.first)][0]++;
   }
   for(auto i = res.begin();i != res.end();){   //   这里 整个帮派的权重和 一定等于 各个结点的权重和 / 2 
       if(i->second[1] / 2 <= k or i->second[0] <= 2) i = res.erase(i);
       else i++;
   }
   cout<<res.size()<<"\n";
   for(auto& i : res){
       cout<<i.first<<" "<<i.second[0]<<"\n";
   }
   return 0;
}