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

题目链接

题意：n个城市，m条路。选出最少里程的若干条路使每个城市都与首都连通。

最小生成树模板题。

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

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

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;
    }
}