hdu 2122 Ice_cream’s world III 最小生成树prim

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define N 1100
#define INF 0x7ffffff
using namespace std;

int n,m,v
,mp

,d
;

int prim()
{
for(int i=0;i<n;i++)    d[i]=INF,v[i]=0;
d[0]=0;
for(int i=0;i<n;i++)
{
int mmin=INF,t;
for(int j=0;j<n;j++)
if(!v[j]&&mmin>d[j])
mmin=d[t=j];
v[t]=1;
for(int j=0;j<n;j++)
if(!v[j]&&mp[t][j]<d[j])
d[j]=mp[t][j];
}
int ans=0;
for(int i=0;i<n;i++)
{
ans+=d[i];
if(ans>=INF)    return -1;
}
return ans;
}

int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
mp[i][j]=INF;
for(int i=0;i<m;i++)
{
int u,v,t;
scanf("%d%d%d",&u,&v,&t);
if(mp[u][v]>t)  mp[u][v]=mp[v][u]=t;
}
int ans=prim();
if(ans==-1)  cout<<"impossible"<<endl;
else    cout<<ans<<endl;
cout<<endl;
}
}