【题解】【AcWing】1518. 团伙头目

最新推荐文章于 2022-02-25 21:35:04 发布

Tuenity

最新推荐文章于 2022-02-25 21:35:04 发布

阅读量166

点赞数

分类专栏： # AcWing

本文链接：https://blog.csdn.net/qq_43733499/article/details/119881255

版权

AcWing 专栏收录该内容

170 篇文章 6 订阅

订阅专栏

1518. 团伙头目

原题传送：AcWing 1518. 团伙头目

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

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

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

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

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

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

输入格式

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

接下来 $N$ 行，每行采用如下形式描述通话：

Name1 Name2 Time

Name1和Name2是通话的两人的名字，Time是通话时间。

名字的长度固定为 $3$ ，且只包含大写字母。

时间是一个不超过 $1000$ 的正整数。

输出格式

第一行输出帮派数量。

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

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

数据范围

$\le N,K \le 1000$

输入样例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

输出样例1：

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

输出样例2：

思路：

dfs查找同一团伙并统计团伙的权值，因为是无向图，所以一条边计算了两遍，需要将权值除以2。

题解：

#include <bits/stdc++.h>

using namespace std;

int n, k;
unordered_map<string, vector<pair<string, int>>> g;
unordered_map<string, int> total;
unordered_map<string, bool> st;

int dfs(string ver, vector<string> &nodes)
{
	st[ver] = true;
	nodes.push_back(ver);
	
	int sum = 0;
	for(auto edge : g[ver])
	{
		sum += edge.second;
		string cur = edge.first;
		if(!st[cur])
			sum +=  dfs(cur, nodes);
	}
	
	return sum;
}

int main()
{
	scanf("%d%d", &n, &k);
	
	while(n--)
	{
		char a[4], b[4];
		int t;
		scanf("%s%s%d", 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 item : total)
	{
		string ver = item.first;
		vector<string> nodes;
		int sum = dfs(ver, nodes) / 2;
		
		if(nodes.size() > 2 && sum > k)
		{
			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());
	
	printf("%d\n", res.size());
	for(auto item : res)
		printf("%s %d\n", item.first.c_str(), item.second);

    return 0;
}