【1034】Head of a Gang (30 分)

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string>
#include<algorithm>  
#include<map>
#include<vector>
#include<queue> 
using namespace std;  
const int maxn=2010; //总人数

map<int, string> intToString; //编号-》姓名
map<string,int> stringToInt;  //姓名-》编号
map<string,int> Gang;  //head->人数
int G[maxn][maxn]={0};
int weight[maxn]={0};  //邻接矩阵G、点权weight
int n,k,numPerson=0; //边数n、下限k及总人数numPerson
bool vis[maxn]={false};  //标记是否被访问

//DFS函数访问单个连通块,nowVisit为当前访问的标号
//head为头目，numMember为成员编号,totalValue为连通块的总边权
void DFS(int nowVisit,int& head, int& numMember,int& totalValue){
	numMember++;//成员人数加1
	vis[nowVisit]=true; //标记nowVisit已访问
	if(weight[nowVisit]>weight[head]){ 
		head=nowVisit; //当前访问结点的点权大于头目的点权，则更新头目
	}
	for(int i=0;i<numPerson;i++){ //枚举所有人
		if(G[nowVisit][i]>0){  //如果从nowVisit能到达i
			totalValue += G[i][nowVisit] ; //连通块的总边权增加该边权
			G[nowVisit][i]=G[i][nowVisit]=0;  //删除这条边,防止回头
			if(vis[i] == false){ //如果i未被访问，则递归访问i
				DFS(i,head,numMember,totalValue);
			}
		}
	}
}

//DFSTrave函数遍历整个图，获取每个连通块的信息
void DFSTrave(){ 
	for(int i=0;i<numPerson;i++){ //枚举所有人
		if(vis[i]==false){  //如果i未被访问
			int head=i,numMember=0,totalValue=0; //头目、成员数、总边权
			DFS(i,head,numMember,totalValue); //遍历i所在的连通块
			//如果成员数大于2且总边权大于k
			if(numMember>2 && totalValue >k){
				//head的人数为numMember
				Gang[intToString[head] ]=numMember;
			}
		}
	}
}

//change函数返回姓名str对应的编号
int change(string str){
	if(stringToInt.find(str) != stringToInt.end()){ //如果str已经出现过
		return stringToInt[str]; //返回编号
	}else{
		stringToInt[str]=numPerson; //str的编号为numPerson
		intToString[numPerson]=str; //numPerson对应str
		return numPerson++;  //总人数加1
	}
}
 
int main(){   
	int w;
	string str1,str2;
	cin >> n>>k;
	for(int i=0; i<n; i++){ 
		cin >> str1>>str2>>w; //输入边的两个端点和点权
		int id1=change(str1);  //将str1转换为编号id1
		int id2=change(str2); //将str2转换为编号id2
		weight[id1]+=w;  //id1的点权增加w
		weight[id2]+=w;  //id2的点权增加w
		G[id1][id2]+=w;  //边id1-》id2的边权增加w
		G[id2][id1]+=w; //边id2-》id1的边权增加w
	}
	DFSTrave();  //遍历整个图的所有连通块,获取Gang的信息
	cout <<Gang.size()<< endl; //Gang的个数
	map<string,int>::iterator it;
	for(it=Gang.begin(); it!=Gang.end();it++){ //遍历所有Gang
		cout << it->first  << " "<< it->second<<endl; //输出信息
	}
	system("pause"); 
    return 0;   
}