【1034】Head of a Gang (30 分)

#include
#include
#include
#include
#include
#include  
#include
#include
#include 
using namespace std;  
const int maxn=2010; //总人数

map intToString; //编号-》姓名
map stringToInt;  //姓名-》编号
map 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;i0){  //如果从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;i2 && 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> 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 <::iterator it;
	for(it=Gang.begin(); it!=Gang.end();it++){ //遍历所有Gang
		cout << it->first  << " "<< it->second<

 

你可能感兴趣的:(PAT)