hdu 1102 Constructing Roads 最小生成树prim模板题

题目链接

题意：有n个村庄，给出n*n的矩阵表示村庄之间的距离。已经建造了Q条路，问建造的最短距离使所有的村庄连通。

最小生成树模板题。

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

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

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]&&mp[t][j]<d[j])
                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))
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&mp[i][j]);
        cin>>m;
        for(int i=0;i<m;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            mp[u][v]=mp[v][u]=0;
        }
        cout<<prim()<<endl;
    }
}