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);//增加联系人b
m[a].time += time;//累加通话时间
m[b].contact.insert(a);//增加联系人a
m[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);//利用并查集，分为不同的集合gang
contactTime[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;//通话时间为0
int 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 gang
head = 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;
}