1034 Head of a Gang (30 分) PAT甲级（图的遍历，DFS）

1034 Head of a Gang (30 分) PAT甲级（图的遍历，DFS）

一、题目
原题是英文且篇幅较长，故不展示，其链接如下：
https://pintia.cn/problem-sets/994805342720868352/problems/994805456881434624
本题主要考察图的遍历(也可以并查集解决)。其在PAT甲级中0.31的通过率，不算太难，但仍有一定复杂度。笔者大二转专业来到计算机学院，之前没有任何基础，在数据结构与算法的学习上始终不得要领，碰到稍微复杂一点的题要么放弃，要么参考题解，边理解边敲一遍别人的代码。寻寻觅觅，兜兜转转，进步甚微。这道题是笔者思考良久，删删改改做出来的（是的，笔者作为即将升入大三的计科学子，水平很差），我认为值得在自己的博客上记录一下，聊以自慰，仅此而已。

二、题意分析
题目给出了若干个人之间的通话记录（看做无向边）。根据通话记录，将他们分为若干组，每个组的总边权即该组内所有通话的长度之和，每个人的点权是该人参与通话的时长之和。现在给定一个阈值K，如果一个组的总边权超过K，且成员人数大于2，将该组视为“犯罪团伙”，并将组内通话时间最长的成员视为头目。要求输出犯罪团伙的个数，并按头目姓名字典序从小到大输出每个“犯罪团伙”的头目姓名和成员人数。
样例1构造的图如下：
三个连通分量的总边权分别为70、75、60,均大于阈值59，但第二个连通分量只有DDD和EEE两个成员，故满足犯罪团伙定义的只有第一个和第三个连通分量。

三、思路整理
1、首先，将信息记录在恰当的数据结构中，成员姓名与编号的对应关系用map记录，每一个成员的通话时长用数组w[]记录，图的结构与边权用邻接矩阵G[][]记录。输出的“犯罪团伙”信息，集成在结构体gang中 ，其中包含了犯罪团伙的人数以及头目的姓名。
以下是信息存储的代码：

int n, least, id = 1;//n是待输入的记录的条数，least是成为犯罪团伙最低的总边权(即通话总时长),id是成员的编号
map<string, int> strToint;//map类型，实现某成员编号到姓名的映射
map<int, string> intTostr;//同理，实现某成员姓名到编号的映射
int w[2010] = { 0 };//记录成员的通话时长（即结点的权值）,w[i]为编号为i的成员通话的总时长
int G[2010][2010] = { 0 };//图的邻接矩阵，如果成员u,v有通话时长，则G[u][v]>0
int vis[2010] = { 0 };//DFS遍历时，开始每个成员均未访问过
struct gang {//犯罪团伙信息结构体
	string boss;//头目的姓名
	int number = 0;//团伙的人数
};
vector<gang> Gang;//保存结果的vector数组，最后遍历输出即可

2、DFS函数的编写。大体思路是遍历所有结点，如果结点未被访问，则进行DFS，可得到一个连通分量，再根据连通分量的信息是否符合犯罪团伙条件，将其加入到Gang数组中。注意，图中可能含有环，在记录连通分量的总边权时，需要将边权加两遍，否则会漏掉一条边的权值。
两种方式代码上的区别（请看代码的详细说明）：

if (G[u][v] > 0) { totalTime += G[u][v]; }
if (vis[v] == false && G[u][v] > 0) {
	DFS(v, bossTime, g, totalTime);
	}

以上的写法：对每个顶点只要相邻的边就加上其边权，最后得到的会是总边权的两倍。对于样例中的FFF,GGG,HHH连通分量，最后得到的总边权就是120。

if (vis[v] == false && G[u][v] > 0) {
	totalTime += G[u][v];
	DFS(v, bossTime, g, totalTime);
	}

以上的写法：只有未被访问的邻边才会加入到总边权，最后会漏掉一条边权。对于样例中的FFF,GGG,HHH连通分量，最后得到的总边权可能是50（取决于最后遍历到的是哪个结点）。

四、代码实现
以下代码有很详细的注释，不足的地方恳请斧正。

#define _CRT_SECURE_NO_WARNINGS
#include<bits/stdc++.h>
using namespace std;
int n, least, id = 1;//n是待输入的记录的条数，least是成为犯罪团伙最低的总边权(即通话总时长),id是成员的编号
map<string, int> strToint;//map类型，实现某成员编号到姓名的映射
map<int, string> intTostr;//同理，实现某成员姓名到编号的映射
int w[2010] = { 0 };//记录成员的通话时长（即结点的权值）,w[i]为编号为i的成员通话的总时长
int G[2010][2010] = { 0 };//图的邻接矩阵，如果成员u,v有通话时长，则G[u][v]>0
int vis[2010] = { 0 };//DFS遍历时，开始每个成员均未访问过
struct gang {//犯罪团伙信息结构体
	string boss;//头目的姓名
	int number = 0;//团伙的人数
};
vector<gang> Gang;//保存结果的vector数组，最后遍历输出即可
bool cmp(gang a, gang b) {//比较函数，最后在输出时要求头目姓名按字典序从小到大
	return a.boss < b.boss;
}
void DFS(int u, int bossTime, gang& g, int& totalTime) {//访问单个结点
	vis[u] = true;//将当前访问到的结点标记
	g.number++;//连通分量的结点数加一
	if (w[u] > bossTime) {//如果当前结点的权值大于之前保存的头目权值，则更新
		bossTime = w[u];
		g.boss = intTostr[u];
	}
	for (int v = 1; v < id; v++) {//遍历所有结点
		if (G[u][v] > 0) { totalTime += G[u][v]; }//加上所有与u相邻的边权，这样会使得到的总边权扩大一倍
		if (vis[v] == false && G[u][v] > 0) {//如果存在相邻且未被访问的结点，对其进行DFS
			DFS(v, bossTime, g, totalTime);
		}
	}
}
void DFSTrave() {//对整张图进行访问
	for (int u = 1; u < id; u++) {//遍历所有的结点
		if (vis[u] == false) {//如果结点未被访问，则对其进行遍历可以得到一个连通分量
			gang g;//g保存连通分量的信息
			int totalTime = 0;//总边权
			g.boss = intTostr[u];//先假定第一个遍历到的是头目
			DFS(u, w[u], g, totalTime);
			if (totalTime > 2 * least && g.number > 2) {//若符合团伙条件，加入数组，由于得到的边权扩大了一被，阈值least也要扩大一倍
				Gang.push_back(g);
			}
		}
	}
}
int main()
{
	scanf("%d%d", &n, &least);//输入记录数目，阈值least
	string name1, name2;//保存每条记录中的姓名
	int weight;//保存每条记录中的边权
	for (int i = 0; i < n; i++) {//输入n条记录
		cin >> name1 >> name2 >> weight;
		if (strToint.find(name1) == strToint.end()) {//为新输入的名字分配一个编号，最后id得到的是最大的成员编号加一
			strToint[name1] = id;
			intTostr[id] = name1;
			id++;
		}
		if (strToint.find(name2) == strToint.end()) {//为新输入的名字分配一个编号，最后id得到的是最大的成员编号加一
			strToint[name2] = id;
			intTostr[id] = name2;
			id++;
		}
		w[strToint[name1]] += weight;//更新结点点权值
		w[strToint[name2]] += weight;//更新结点权值
		G[strToint[name1]][strToint[name2]] += weight;//更新图的边权
		G[strToint[name2]][strToint[name1]] += weight;//更新图的边权
	}
	DFSTrave();//图的遍历，得到符合条件的犯罪团伙存储在Gang中
	sort(Gang.begin(), Gang.end(), cmp);//按头目的字典序升序排列
	printf("%d\n", Gang.size());
	for (int i = 0; i < Gang.size(); i++) {//输出答案
		cout << Gang[i].boss << " " << Gang[i].number << endl;;
	}
	return 0;
}