PAT A1034 Head of a Gang (30 分)

最新推荐文章于 2024-09-13 16:27:15 发布

柠檬猫w

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量72

点赞数

文章标签：算法 dfs

本文链接：https://blog.csdn.net/weixin_45964844/article/details/113700842

版权

博主分享了在学习图论过程中遇到的一道图的遍历题目PAT A1034，尽管独立完成有一定挑战，但通过参考和调试，期望在后续练习中不断提升。

摘要由CSDN通过智能技术生成

在这里插入图片描述
刚刚开始学习图论，这是一道图的遍历题，目前对我来说独立完成还是有点困难的。。。自己想了很久，参考了书上的代码，也调试了很久，希望做后面的题目能有进步吧。

#include <cstdio>
#include <vector>
#include <map>
#include <iostream>
#include <algorithm>

using namespace std;

int n, k;
int bh = 0;
map<string,int> ntob;
map<int,string> bton;
map<string,int> cishu;
int G[2020][2020] = {0};
int weight[2020] = {0};
bool visited[2020] = {false};

struct Gang{
	string name;
	int num;
}gang[1010];

bool cmp(Gang g1, Gang g2){
	return g1.name < g2.name;
}

int change(string s){
	if(cishu[s] == 0){
		cishu[s]++;
		ntob[s] = bh++;
		return bh-1;
	}
	return ntob[s];
}

void DFS(int index, int &head, int &sum, int &num){
	visited[index] = true;
	num++;
	if(weight[index] > weight[head]){
		head = index;
	}
	for(int i=0; i<bh; i++){
		if(G[index][i] != 0){
			sum += G[index][i];
			G[index][i] = G[i][index] = 0;
			if(visited[i] == false){
				DFS(i, head, sum, num);
			}
		}
	}
}

int main(){
	scanf("%d %d", &n, &k);
	for(int i=0; i<n; i++){
		string s1, s2;
		int w;
		cin >> s1 >> s2 >> w;
		int bh1 = change(s1);
		int bh2 = change(s2);
		bton[bh1] = s1;
		bton[bh2] = s2;
		G[bh1][bh2] += w;
		G[bh2][bh1] += w;
		weight[bh1] += w;
		weight[bh2] += w;
	}
	
	int g = 0;
	for(int i=0; i<bh; i++){
		if(visited[i] == false){
			int head = i;
			int sum_w = 0;
			int w = weight[i];
			int num = 0;
			DFS(i, head, sum_w, num);
			if(sum_w>k && num>2){
				gang[g].name = bton[head];
				gang[g].num = num;
				g++;
			}
		}
	}
	
	sort(gang, gang+g, cmp);
	
	printf("%d\n", g);
	for(int i=0; i<g; i++){
		printf("%s %d\n", gang[i].name.c_str(), gang[i].num);
	}
	
	return 0;
}