PAT甲级1034 Head of a Gang (30 分)题解

最新推荐文章于 2022-11-08 11:47:45 发布

程勇uestc

最新推荐文章于 2022-11-08 11:47:45 发布

阅读量1.4k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/qq_40438165/article/details/100340691

PAT甲级专栏收录该内容

96 篇文章

订阅专栏

本文介绍了一种使用图遍历算法解决复杂团队通话数据分析问题的方法，通过姓名到索引的映射建立图结构，记录每个人及团队的通话总时间，采用深度优先搜索(DFS)找出所有通话团体，并筛选出合法团队。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
$\quad$ 这个题虽然只是图的遍历问题，但写程序要求较高，几点需要注意

给出的人员姓名不是数字是字符串，需要建立姓名到索引和索引到姓名的映射，方便建图和后续处理
需要记录每个人通话总时间，团队通话总时间就是该团队内所有人通话时间和的一半
1000条通话记录最多可能有2000人，故需要开辟大于2000的空间，不然有个测试点会显示段错误

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;

const int maxn = 2001; // 1000条记录最多2000人
vector<int> E[maxn];
int call[maxn];  // 记录每个人通话时间
int vis[maxn];
struct Gang{ //记录每一个团伙信息
	string head;
	int num=0, totalCall=0, flag=0;
}gang[maxn];
int callMax = 0;
map<string, int> mp;  // 姓名到编号的索引
map<int, string> rmp;  // 编号到姓名的索引
void dfs(int u, Gang &g)
{
	vis[u] = 1;
	g.num++, g.totalCall+=call[u];  // 更新团伙信息
	if(call[u]>callMax) callMax=call[u], g.head=rmp[u];
	for (int i = 0; i < E[u].size(); ++i){
		int v = E[u][i];
		if(vis[v]==0) dfs(v, g);
	}
}
bool cmp(const Gang &g1, const Gang &g2) { return g1.head<g2.head; }
int main(int argc, char const *argv[])
{
	int N, K; cin >> N >> K;
	int num = 0;
	for (int i = 0; i < N; ++i){
		string a, b;
		int time;
		cin >> a >> b >> time;
		if(mp.count(a)==0) rmp[num]=a, mp[a] = num++;
		if(mp.count(b)==0) rmp[num]=b, mp[b] = num++;
		call[mp[a]] += time, call[mp[b]] += time;  // 更新通话时间
		E[mp[a]].push_back(mp[b]);
		E[mp[b]].push_back(mp[a]);
	}
	int index = 0;
	for (int i = 0; i < num; ++i){
		callMax = 0;  // 每一次dfs前置最大通话时间为0 
		if(vis[i]==0) dfs(i, gang[index++]);
	}
	sort(gang, gang+index, cmp);  // 按照领头人姓名排序
	int res = 0;  // 记录合法的团伙数
	for (int i = 0; i < index; ++i){
		if(gang[i].num<=2 || gang[i].totalCall/2<=K) continue;  // 不合法
		gang[i].flag=1, res++;
	}
	cout << res << endl;
	for (int i = 0; i < index; ++i){
		if(gang[i].flag==1) cout << gang[i].head << " " << gang[i].num << endl;
	}
	return 0;
}