PAT A1034

最新推荐文章于 2022-09-11 16:00:00 发布

原创最新推荐文章于 2022-09-11 16:00:00 发布 · 483 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT 甲级

PAT 甲级专栏收录该内容

130 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索(DFS)进行图遍历的方法，该方法用于寻找符合特定条件的连通分量，包括节点数量超过两个且权重总和大于给定阈值的分量。文章详细展示了如何通过C++实现这一算法，并处理了大型数据集中的大量唯一标识符。

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

方法：用深度遍历，同时计算每个连通分量的结点数，及点权最大的结点。

注意点：1，输出的连通分量要求，结点数>2且点权之和大于指定值；2，题中有至多1000个记录，但至多会有2000个不同的名字。

#include <iostream>
#include <string>
#include <map>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn=2010;
map<string,int> mp;
map<int,string> mp2;  //数字到名字的映射 
vector<int> child[maxn];
int N,K,weight[maxn]={0},index=0;  //index记录所有电话记录包含的人数 

map<string,int> gang; 
bool vis[maxn]={0};
void DFS(int v,int &num,int &headindex,int &sumw,int &maxw){
	vis[v]=true;
	num++;
	sumw+=weight[v];
	if(weight[v]>maxw){
		headindex=v;
		maxw=weight[v];
	}
	for(int i=0;i<child[v].size();i++){
		if(vis[child[v][i]]==false){
			DFS(child[v][i],num,headindex,sumw,maxw);
		}
	}
}
void DFStrav(){
	int numgang=0;
	for(int i=0;i<index;i++){
		if(vis[i]==false){
			int num=0,headindex=-1,sumw=0,maxw=-1;
			DFS(i,num,headindex,sumw,maxw);
			if(num<=2||sumw<=2*K) continue;
			gang[mp2[headindex]]=num;
			numgang++;
		}
	}
	printf("%d\n",numgang);
}
int main(){
	scanf("%d%d",&N,&K);
	string s1,s2;
	int time;
	for(int i=0;i<N;i++){
		cin>>s1>>s2>>time;
		if(mp.find(s1)==mp.end()){
			mp[s1]=index++;
			mp2[index-1]=s1;
		}
		weight[mp[s1]]+=time;
		if(mp.find(s2)==mp.end()){
			mp[s2]=index++;
			mp2[index-1]=s2;
		}
		weight[mp[s2]]+=time;
		child[mp[s1]].push_back(mp[s2]);
		child[mp[s2]].push_back(mp[s1]);
	}
	DFStrav();
	for(map<string,int>::iterator it=gang.begin();it!=gang.end();it++){
		cout<<it->first<<" "<<it->second<<endl;
	}
	return 0;
}