LightOJ-1041-prime

题目大意：有若干个城市，给出m条路，如果cost不是0就代表这条路需要修，问要使这个图形成连通图，最少需要花费多少；

题目解析：最小生成树，prime，因为不知道有多少个城市，先用map来映射一下；

AC代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
using namespace std;
const int inf=0x3fffffff;
map<string,int>ma;
int graph[101][101],dist[101],n;
bool vis[101];
int prime()
{
	int i,j,flag,temp,sum=0,t;
	memset(vis,0,sizeof(vis));
	vis[1]=1;
	for(i=1;i<=n;i++)
	{
		dist[i]=graph[1][i];
	}
	flag=1;
	for(i=1;i<=n-1;i++)
	{
		temp=inf;
		t=-1;
		for(j=1;j<=n;j++)
		{
			if(!vis[j]&&temp>dist[j])
			{
				temp=dist[j];
				t=j;	
			}
		}
		if(t==-1)
		{
			flag=0;
			break;
		}
		vis[t]=1;
		sum+=dist[t];
		for(j=1;j<=n;j++)
		{
			if(!vis[j])
			{
				if(graph[t][j]<dist[j])
					dist[j]=graph[t][j];
			}
		}
	}
	if(flag==0)
		return -1;
	else 
		return sum;
}
int main()
{
	int c,cas,m,i,w,temp,j;
	string u,v;
	scanf("%d",&cas);
	for(c=1;c<=cas;c++)
	{
		for(i=1;i<=100;i++)
			for(j=1;j<=100;j++)
				graph[i][j]=inf;
		ma.clear();
		scanf("%d",&m);
		n=0;
		for(i=0;i<m;i++)
		{
			cin>>u>>v;
			if(ma[u]==0)
				ma[u]=++n;
			if(ma[v]==0)
				ma[v]=++n;
			scanf("%d",&w);
			graph[ma[u]][ma[v]]=graph[ma[v]][ma[u]]=min(graph[ma[v]][ma[u]],w);
		}		
		temp=prime();
		if(temp==-1)
		printf("Case %d: Impossible\n",c);
		else 
		printf("Case %d: %d\n",c,temp);
	}
	return 0;
}