题意:有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;
}
}