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

题目链接

题意：给出n个点，n*(n-1)/2条路，求最少的距离将所有点连接起来。

最小生成树模板题。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define N 110
#define INF 0x7ffffff

using namespace std;

int v[N],d[N],mp[N][N],n;

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

int main()
{
    while(~scanf("%d",&n)&&n)
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                mp[i][j]=INF;
        int m=n*(n-1)/2;
        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;
        }
        cout<<prim()<<endl;
    }
}