hdu 1233 还是畅通工程(最小生成树)

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

const int INF=0x3f3f3f3f;
const int MAXN=110;
bool vis[MAXN];
int lowc[MAXN];

int Prim(int cost[][MAXN],int n)
{
int ans=0;
memset(vis,false,sizeof(vis));
vis[0]=true;
for(int i=1; i<n; i++)lowc[i]=cost[0][i];
for(int i=1; i<n; i++)
{
int minc=INF;
int p=-1;
for(int j=0; j<n; j++)
if(!vis[j]&&minc>lowc[j])
{
minc=lowc[j];
p=j;
}
if(minc==INF)return -1;//原图不连通
ans+=minc;
vis[p]=true;
for(int j=0; j<n; j++)
if(!vis[j]&&lowc[j]>cost[p][j])
lowc[j]=cost[p][j];
}
return ans;
}

int main()
{
int cost[MAXN][MAXN];
int N;
int u,v,w;
while(~scanf("%d",&N))
{
if(N==0)break;
for(int i=0; i<MAXN; ++i)
{
for(int j=0; j<MAXN; ++j)
{
cost[i][j]=INF;
}
}

int M=N*(N-1)/2;
for(int i=0; i<M; ++i)
{
scanf("%d%d%d",&u,&v,&w);
--u;
--v;
cost[u][v]=cost[v][u]=w;
}

printf("%d\n",Prim(cost,N));
}
return 0;
}