HDU 1233 还是畅通工程 // prime() 算法

HDU 1233  还是畅通工程

prime()算法，计算从1到n的最小总长度

//prime算法 
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<set>
#include<vector>
using namespace std;
#define INF 0x3f3f3f3f 
int g[200][200];
int n,m;
int dis[200],vis[200];
int prime()
{
	int minn,temp;
	int sum=0;
	for(int i=1;i<n;i++)
	{
		minn=INF;
		for(int j=1;j<=n;j++)
		{
			if(vis[j]==0 && dis[j]<minn)
			{
				minn=dis[j];
				temp=j;
			}
		}
		vis[temp]=1;
		sum+=minn;
		for(int j=1;j<=n;j++)
		{
			if(vis[j]==0 && dis[j]>g[temp][j])
			{
				dis[j]=g[temp][j];
			}
		}
	}
	return sum;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
    	if(n==0)
    	 break;
	    m=(n*(n-1))/2;
	    memset(vis,0,sizeof(vis));
	    memset(dis,0,sizeof(dis));
	    for(int i=1;i<=n;i++)
	    {
	    	for(int j=1;j<=n;j++)
	    	{
	    		if(i==j)
	    		g[i][j]=0;
	    		else
	    		g[i][j]=INF;
			}
		}
		for(int i=1;i<=m;i++)
		{
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			g[u][v]=g[v][u]=min(g[u][v],w);
		}
		for(int i=1;i<=n;i++)
		{
			dis[i]=g[1][i];
		}
		dis[1]=0;
		vis[1]=1;
		cout<<prime()<<endl;	
	}
 return 0;
}