1034 Head of a Gang （30 point(s)）

深度优先搜索在社交网络分析中的应用

最新推荐文章于 2025-04-08 00:38:28 发布

徐伯莱

最新推荐文章于 2025-04-08 00:38:28 发布

阅读量196

点赞数

CC 4.0 BY-SA版权

分类专栏： Pat甲级文章标签： Pat甲级

本文链接：https://blog.youkuaiyun.com/m0_37691414/article/details/86681274

Pat甲级专栏收录该内容

152 篇文章

订阅专栏

本文探讨了深度优先搜索（DFS）在解决社交网络中特定问题的应用，如寻找影响力最大的个体及其社群规模。通过构建图模型，利用DFS遍历节点，匹配字符串与整数，统计社群内权重总和及成员数量，筛选出满足特定条件的社群领袖。

题解

dfs。技巧性在于string与int之间的匹配，毕竟dfs只能遍历节点呀。你说对不对。

#include<iostream>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
const int MAXN = 2e3 + 10; //1000 * 2
vector<string> name; // 人名 
map<string, int> idcache;
int n, k;
int e[MAXN][MAXN], w[MAXN];
bool vis[MAXN];
int ID(string x) {
	if(idcache.count(x)) return idcache[x];
	name.push_back(x);
	return idcache[x] = name.size() - 1;
} 
void dfs(int u, int& head, int &num, int &totalw) {
	vis[u] = true;
	++num;
	if(w[u] > w[head]) head = u;
	for(int v = 0; v < name.size(); ++v) {
		if(e[u][v]) {
			totalw += e[u][v];
			e[u][v] = e[v][u] = 0;
			if(vis[v] == false) dfs(v, head, num, totalw);
		}
	} 
}
map<string, int> ret;
void solve() {
	for(int i = 0; i < name.size(); ++i) if(vis[i] == false) {
		int head = i, totalw = 0, num = 0;
		dfs(i, head, num, totalw);
		if(num > 2 && totalw > k) ret[name[head]] = num;
	}
}
int main() {
	scanf("%d%d", &n, &k);
	int t;
	string a, b;
	for(int i = 0; i < n; ++i) {
		cin >> a >> b >> t;
		int aa = ID(a), bb = ID(b);
		e[aa][bb] += t, e[bb][aa] += t;
		w[aa] += t, w[bb] += t;
	}
	solve();
	printf("%d\n", ret.size());
	for(map<string, int>::iterator it = ret.begin(); it != ret.end(); ++it) {
		cout << it->first << " " << it->second << '\n';
	}
	return 0;
}