【PAT甲级】1034 Head of a Gang (30 分)

一、题目分析

1. 翻译

gang：黑帮，犯罪团伙
threshold：门槛

2. 分析

1）题意：A和B之间有通话，即A和B有联系；A和B之间联系的权值为A和B通话时间的总和；Gang是指一个集群，集群中超过两个人，且每两个人互相之间的通话时间之和要超过给出的门槛值K；在每个Gang中拥有最大权值的结点即为head；每个结点的权值是它所连接的边的权值之和。给出结点总数 通话关系数N和门槛值K，以及N个通话关系，要求找出存在几个Gang，分别输出它们的head和集群中成员的总人数。
2）N是关系总数，而不是结点总数，故设置结点数最大值MAXV须大于2000，以防段错误。
3）1️⃣每个人的名称及其序号之间的映射用map实现，map在我理解中可以视作一个“数组”，元素是一对一对出现的。2️⃣map<typename1, typename2> mp表示typename1 到typename2 的映射，mp[typename1]=typename2，注意是[]，而不是()。3️⃣mp.end()指向最后一对元素的下一对元素的位置，即为空，所以可以用此判断mp中是否有某一对元素。
4️⃣mp.find(typename1)返回的是一个迭代器类型，可以理解成指向<typename1,typename2>的一个指针。5️⃣map<typename1,typename2>::iterator it设置出一个迭代器，可以使用it->first和it->second分别访问元素typename1和typename2。
4）使用深度优先搜索，切记模板。
5）此题构成的图为有向图，但是边权可以按照无向图处理，例如A->B=10，B->A=20，则可以直接记A-B=30。那么，在计算完Gang中边权总和时，切记删去A->B和B->A两条路径。
6）Gang要满足两个条件：2人以上且边权总和大于K.

二、代码解析

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <string>
#include <cstring>
#include <cctype>
#include <unordered_map>
#include <stack>
#include <queue>
using namespace std;
//n为通话数，每次通话有两个人，因此人数（结点数）不少于2000
const int MAXV=2010;
map<string,int> stringToint;
map<int,string> intTostring;
map<string,int> Gang;//head->人数
int n,k,numPerson=0;
int G[MAXV][MAXV]= {0},weight[MAXV]= {0};
bool visit[MAXV]= {false};
int change(string str) {
	if(stringToint.find(str)!=stringToint.end()) {
		return stringToint[str];// [] not ()
	} else {
		stringToint[str]=numPerson;
		intTostring[numPerson]=str;
		return numPerson++;
	}
}
void DFS(int nowVisit,int &head,int &numMember,int &totalValue) {//引用
	numMember++;
	visit[nowVisit]=true;
	if(weight[nowVisit]>weight[head]) {
		head=nowVisit;
	}
	for(int i=0; i<n; i++) {
		if(G[nowVisit][i]>0) {//只要有路径，就需要更新totalValue
			totalValue+=G[nowVisit][i];
			G[nowVisit][i]=G[i][nowVisit]=0;
			if(!visit[i]) {//当i未访问过才进行DFS
				DFS(i,head,numMember,totalValue);
			}
		}
	}
}
void dfsTrave() {
	for(int i=0; i<n; i++) {//n=numPerson
		if(!visit[i]) {
			int head=i,numMember=0,totalValue=0;
			DFS(i,head,numMember,totalValue);
			if(totalValue>k&&numMember>2) {
				Gang[intTostring[head]]=numMember;
			}
		}
	}
}
int main() {
	cin>>n>>k;
	for(int i=0; i<n; i++) {
		int u,v,w;
		string a,b;
		cin>>a>>b>>w;
		u=change(a);
		v=change(b);
		weight[u]+=w;
		weight[v]+=w;
		//边权值按照无向图处理，表示这种“关系”的权值
		//且防止走回头路时，来回两条路径均会变成不可达
		G[u][v]+=w;
		G[v][u]+=w;
	}
	dfsTrave();
	cout<<Gang.size()<<endl;
	map<string,int>::iterator it;
	for(it=Gang.begin(); it!=Gang.end(); it++) {
		cout<<it->first<<" "<<it->second<<endl;
	}
	return 0;
}

三、我的疑问

无

如果喜欢我的博客，欢迎点赞、评论、收藏~~谢谢
( * ^ ▽ ^ * ) ~