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

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

Description
当前农村公路建设正如火如荼的展开，某乡镇政府决定实现村村通公路，工程师现有各个村落之间的原始道路统计数据表，表中列出了各村之间可以建设公路的若干条道路的成本，你的任务是根据给出的数据表，求使得每个村都有公路连通所需要的最低成本。
Input
连续多组数据输入，每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000)，随后M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本，村庄从1～N编号。
Output
输出使每个村庄都有公路连通所需要的最低成本，如果输入数据不能使所有村庄畅通，则输出-1，表示有些村庄之间没有路连通。
Samples
Sample #1
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
Output
19

分析：

设顶点集A为最小生成树的点集；顶点集B为待处理点集
a[i]用来保存，顶点集A到i顶点的最小代价。
a数组初始化为INF（无穷大）
随着结点的不断并入，lowcost数组的值不断更新
为0代表该结点已并入A顶点集
为INF代表该结点与A顶点集无直接边相连

#include <bits/stdc++.h>
#include <string.h>
using namespace std;
# define INF 0x3f3f3f3f 
const int N=1e6 +10;
int n,m,t,k,s,sum=0,i,j,flag=1,cnt;
int graph[2010][2010],vis[2010],a[2010];
int prim(){
	a[1]=0;
	for(i=2;i<=n;i++)
	    a[i]=graph[i][1];
	for(i=2;i<=n;i++){
		int min=INF;
		j=1,k=0;
		while(j<=n){
			if(a[j]!=0&&a[j]<min){
				min=a[j];
				k=j;
			}
			j++;
		}
		if(min==INF)
		    break;
		sum+=min;
		a[k]=0;
		for(j=1;j<=n;j++){
			if(a[j]!=0&&a[j]>graph[j][k])
			    a[j]=graph[j][k];
		}
	}
	for(i=1;i<=n;i++){
		if(a[i]!=0)
		    return -1;
	}
	return sum;
}
int main(){
   	while(cin>>n>>m){
   		sum=0;
   		memset(graph,INF,sizeof(graph));
   		memset(vis,0,sizeof(vis));
   		memset(a,INF,sizeof(a));
   		int u,v;
		   flag=1,cnt=0;
   		while(m--){//m条边 
   		    cin>>u>>v>>k;
   		    graph[u][v]=graph[v][u]=k;
		}
		cout<<prim()<<endl;
	}
	return 0;
}