1034. Head of a Gang (30)

1.该题目涉及到了并查集和一些统计

2.

过程

1）收集根据记录，存储每个人的总通话时间和他的联系人，同时进行并查集的合并。

2）根据并查集中分好的集合，进行集合的分类，并且分别统计各集合的名单和总通话时间

3）遍历上面的集合，找出符合条件的gang和gang的头目

注：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

有联系即可，不是gang里面每个人都相互联系。

AC的代码如下，利用了大量的map，set之类，有点难懂

//#include<string>//#include <iomanip>#include<vector>#include <algorithm>//#include<stack>#include<set>#include<queue>#include<map>//#include<unordered_set>#include<unordered_map>//#include <sstream>//#include "func.h"//#include <list>#include<stdio.h>#include<iostream>#include<string>#include<memory.h>#include<limits.h>using namespace std;struct guyNode{set<string> contact;int time;string r;guyNode() : time(0), r(""){ contact.clear(); };};string find(string a,map<string, guyNode>&m){if (m[a].r == a)return a;else{m[a].r = find(m[a].r, m);return m[a].r;}}int main(void){int recordNum, threshold;cin >> recordNum >> threshold;map<string, guyNode> m;//存储每个人的总通话时间和联系人for (int i = 0; i < recordNum; i++){//输入联系记录string a, b;int time;cin >> a >> b >> time;if (m[a].r == "") m[a].r = a;//如果是第一次输入，则把代表设为自己if (m[b].r == "") m[b].r = b;m[a].contact.insert(b);//增加联系人bm[a].time += time;//累加通话时间m[b].contact.insert(a);//增加联系人am[b].time += time;//累加通话时间if (find(a,m)!=find(b,m))m[find(a, m)].r = find(b, m);//合并}map<string, set<string>> gang;//存储gang的代表和成员，注意，代表还不是头目map<string, int> contactTime;//计算各个gang的总通话时间，注意，a和b通话，a累加了一次时间，b又累加了一次时间，所以是实际时间的两倍for (map<string, guyNode>::iterator ite = m.begin(); ite != m.end(); ite++){//遍历所有的集合gang[find(ite->first, m)].insert(ite->first);//利用并查集，分为不同的集合gangcontactTime[find(ite->first, m)] += ite->second.time;//统计各个gang的时间}map<string, int> ans;for (map<string, set<string>>::iterator ite = gang.begin(); ite != gang.end(); ite++){if (contactTime[ite->first] <= threshold * 2 || ite->second.size()<=2) continue;//通话时间不满足要求,人数少于两人bool isGang = true;//标志位string head = "";//设置头目信息int maxTime = 0;//通话时间为0int sum = ite->second.size();//gang的总人数for (set<string>::iterator ite2 = ite->second.begin(); ite2 != ite->second.end(); ite2++){string name = *ite2;if (m[name].time > maxTime){//查找通话时间最长的人，即head of ganghead = name;maxTime = m[name].time;}}if (isGang){ans[head] = sum;}}cout << ans.size() << endl;for (map<string, int>::iterator ite = ans.begin(); ite != ans.end(); ite++){//输出gang的head和成员数cout << ite->first << " " << ite->second << endl;}return 0;}