L2-2 团伙头目 (25 分)

警察找到团伙头目的一种方法是检查人们的通话。

如果 A 和 B 之间有通话，我们就说 A 和 B 是相关的。并且关联具有传递性，即如果 A 与 B 关联，B 与 C 关联，那么 A 与 C 也是关联的。

关联权重定义为两人之间所有通话的总时间长度。

一个“帮派”是一个由至少3个相互关联的人组成的群体，并且其总关联权重大于给定的阈值 K。

在每个帮派中，总权重最大的就是头目，数据保证每个帮派中总权重最大的人是唯一的。

你需要确定各个帮派以及帮派头目。

数据范围

1≤N,K≤1000

输入格式

第一行包含两个整数 N 和 K，表示通话数量以及权重阈值。

接下来 N 行，每行采用如下形式描述通话：
Name1 Name2 Time
Name1 和 Name2 是通话的两人的名字，Time 是通话时间。
名字的长度固定为 3，且只包含大写字母。
时间是一个不超过 1000 的正整数。

输出格式

第一行输出帮派数量。

接下来每行输出一个帮派的信息，包括帮派头目名字以及帮派人数。

帮派信息按头目名字字典序输出。

输入样例

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

输出样例

2
AAA 3
GGG 3

输入样例2

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

输出样例2

0

思路

别人的思路，深度优先遍历无向图，找出各个连通域，详情看代码

#include<bits/stdc++.h>
using namespace std;

int n, m;
unordered_map<string, vector<pair<string, int>>> g;
unordered_map<string, int> total;
unordered_map<string, int> vis;
int dfs(string st, vector<string> &nodes)
{
	vis[st] = 1;
	nodes.push_back(st);
	int sum = 0;
	for(auto it : g[st]){
		sum += it.second;
		if(!vis[it.first])
			sum += dfs(it.first, nodes);
	}
	return sum;
}
int main()
{
	cin>>n>>m;
	while(n--){
		string a, b;
		int t;
		cin>>a>>b>>t;
		g[a].push_back({b, t});
		g[b].push_back({a, t});
		total[a] += t;
		total[b] += t;
	}
	vector<pair<string, int>> res;
	for(auto it : total){
		if(vis[it.first])
			continue;
		string st = it.first;
		vector<string> nodes;
		int sum = dfs(st, nodes) / 2;
		if(sum > m && nodes.size() > 2){
			string boss = nodes[0];
			for(string node : nodes)
				if(total[boss] < total[node])
					boss = node;
			res.push_back({boss, nodes.size()});
		}
	}
	sort(res.begin(), res.end());
	cout<<res.size()<<endl;
	for(auto it: res)
		cout<<it.first<<' '<<it.second<<endl;
 }