【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;
      }
  
 

文章来源: andyguo.blog.csdn.net，作者：山顶夕景，版权归原作者所有，如需转载，请联系作者。

原文链接：andyguo.blog.csdn.net/article/details/100022367