SDUT 3362-村村通公路

数据结构实验之图论六：村村通公路

Time Limit: 1000MS Memory Limit: 65536KB

Problem Description

当前农村公路建设正如火如荼的展开，某乡镇政府决定实现村村通公路，工程师现有各个村落之间的原始道路统计数据表，表中列出了各村之间可以建设公路的若干条道路的成本，你的任务是根据给出的数据表，求使得每个村都有公路连通所需要的最低成本。

Input

连续多组数据输入，每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000)，随后M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本，村庄从1～N编号。 

Output

输出使每个村庄都有公路连通所需要的最低成本，如果输入数据不能使所有村庄畅通，则输出-1，表示有些村庄之间没有路连通。 

Example Input

5 8
1 2 12
1 3 9
1 4 11
1 5 3
2 3 6
2 4 9
3 4 4
4 5 6

Example Output

19

这里我用的是普里姆算法

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
#define max 100000
struct graph
{
	int v,c;
	int ma[1001][1001];
}g;
int vis[1001],mi[1001];//mi辅助数组
int prim()
{
	int sum=0,flag,m;
	vis[1]=1;//可以从任意顶点开始，这里从1开始
	for(int i=1;i<=g.v;i++) mi[i]=g.ma[1][i];//初始化
	for(int i=1;i<g.v;i++)
	{
		m=max;
		flag=0;
		for(int j=1;j<=g.v;j++)
		{
			if(mi[j]<m&&!vis[j])//找出与未访问的点中成本最小的
			{
				m=mi[j];
				flag=j;
			}
		}
		if(m==max) return -1;//未找到就说明是非连通的，直接跳出
		vis[flag]=1;
		sum+=m;//成本累加
		for(int j=1;j<=g.v;j++)//更新
		{
			if(!vis[j]&&mi[j]>g.ma[flag][j])
			{
				mi[j]=g.ma[flag][j];
			}
		}
	}
	return sum;
}
int main()
{
	int n,m,a,b,q,result;
	while(cin>>n>>m)
	{
		for(int i=0;i<=n;i++)
		{
			for(int j=0;j<=n;j++)
			{
				if(j==i) g.ma[i][j]=0;
				else
				{
					g.ma[i][j]=g.ma[j][i]=max;
				}
			}
		}
		for(int i=0;i<m;i++)
		{
			cin>>a>>b>>q;
			g.ma[a][b]=g.ma[b][a]=q;
		}
		memset(vis,0,sizeof(vis));
		g.v=n;
		result=prim();
		cout<<result<<endl;
	}
	return 0;
}