图的遍历&团伙头目

前言：在学习遍历图的过程中发现了这道题，用dfs和邻接矩阵实现遍历，还会用到map构造映射关系简化代码

题目：PAT 1034

思路：用邻接矩阵存储无向图（邻接表也可以操作），可以看出图被分成好几个连通块，依次对每个连通块进行遍历得到该连通块中的头目、成员数、边权之和，再进行条件判断是否要存入答案map。

代码如下：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <string>
#include <map>
using namespace std;
const int maxn = 2010;

map<int, string> inttostring;//编号->姓名 
map<string, int> stringtoint;//姓名->编号 
map<string, int> gang;//头目名->其团队人数 

int n,k,numall;//总边数，最低限数，总人数 
int weight[maxn],G[maxn][maxn];//每个点的权值，每条边的权值 
int v[maxn];//标记数组 

int change(string str)//构造姓名与编号的映射
{
	if(stringtoint.find(str) != stringtoint.end())//str已经出现过 
		return stringtoint[str];
	else
	{
		stringtoint[str] = numall;
		inttostring[numall] = str;
		return numall++;//同时更新总人数
	}
}

void dfs(int i, int&head, int&num, int&value)//引用传递
{
	num++;
	v[i]=1;
	if(weight[i] > weight[head])
		head = i;//根据点权更新头目 
	for(int j = 0; j < numall; j++)//枚举所有人 
	{
		if(G[i][j] > 0)
		{
			value += G[i][j];//主意加的是边权值
			G[i][j] = G[j][i] = 0;//加过的边立刻删掉
			if(!v[j]) dfs(j, head, num, value);
		}
	}
}

int main()
{
	cin>>n>>k;
	for(int i=0;i<n;i++)
	{
		string s1,s2;
		int w;
		cin >> s1 >> s2 >> w;
		int id1 = change(s1);
		int id2 = change(s2);
		weight[id1] += w;
		weight[id2] += w;
		G[id1][id2] += w;
		G[id2][id1] += w;
	}
	for(int i=0; i<numall; i++)
	{
		if(!v[i])
		{
			int head = i,num = 0,value = 0;
			dfs(i, head, num, value);
			if(value > k && num > 2)
			{
				gang[inttostring[head]] = num;//头目->该团队人数 
			}
		} 
	}
	cout<< gang.size() <<endl;
	for(map<string, int>::iterator it = gang.begin(); it != gang.end(); it++)//访问map gang中所有成员
	{
		cout << it->first << " " << it->second << endl;//该头目和其团队人数 
	}
	return 0;
}