pat 1034 Head of a Gang

题意是要求总共有几个集合，每个集合权重最大的是哪个，以及每个集合中结点个数。一开始老想用并查集来做，发现实现起来很困难。参考网上代码，发现可以用DFS遍历，数出每个分支的结点个数。其中map的迭代器的使用得注意。还有，若一个gang中同时有多个相同的最大值，取字母序最小的。虽然不加这点，也能过，但为了健壮性，加上吧。

 

AC代码：

#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<map>
using namespace std;

map<string, vector<string> > adj;//邻接表
map<string, int> weight;//各点权重
map<string, int> visit;//是否访问过
map<string, int> ans;//最后要输出的结果
int cnt,totalweight;
string head;

void dfs(string str)
{
	visit[str]=1;
	cnt++;//该集合中结点个数
	totalweight+=weight[str];
	if(weight[str]>weight[head])
		head=str;
	else if(weight[str]==weight[head]&&str<head)//这里得注意，若一个gang中同时有多个相同的最大值，取字母序最小的
		head=str;
	for(vector<string>::iterator it=adj[str].begin();it!=adj[str].end();it++){
		if(visit[*it]==0){

			dfs(*it);

		}
	}
}

int main()
{
	int n,k,time;
	string name1,name2;
	//freopen("C:\\Documents and Settings\\Administrator\\桌面\\input.txt","r",stdin);

	cin>>n>>k;
	int i;
	for(i=0;i<n;i++){
		cin>>name1>>name2>>time;
		weight[name1]+=time;
		weight[name2]+=time;
		adj[name1].push_back(name2);
		adj[name2].push_back(name1);
		visit[name1]=0;
		visit[name2]=0;
	}
	for(map<string, int >::iterator it=visit.begin();it!=visit.end();it++){
		if(visit[it->first]==0){
			cnt=0;//该集合中结点个数
			totalweight=0;//该集合总权重
			head=it->first;
			dfs(it->first);
			if(cnt>2 && totalweight/2 > k)//满足条件则登记
				ans[head]=cnt;
		}
	}

	cout<<ans.size()<<endl;
	for(map<string,int>::iterator iter=ans.begin();iter!=ans.end();iter++){
		cout<<iter->first<<" "<<iter->second<<endl;
	}
	return 0;
}