题意:
有n个农场,已知这n个农场都互相相通,有一定的距离,现在每个农场需要装光纤,问怎么安装光纤能将所有农场都连通起来,并且要使光纤距离最小,输出安装光纤的总距离。
题解:
最小生成树模板。
//prim
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
const int MAXN=105;
int map[MAXN][MAXN];
int dis[MAXN];
bool vis[MAXN];
int n;
void prim()
{
int sum=0;
memset(vis,false,sizeof(vis));
for(int i=1;i<=n;i++)
dis[i]=map[1][i];
vis[1]=true;
for(int i=1;i<=n;i++)
{
int MIN=INF,k=0;
for(int j=1;j<=n;j++)
if(!vis[j]&&MIN>dis[j])
MIN=dis[j],k=j;
vis[k]=true;
for(int j=1;j<=n;j++)
if(!vis[j]&&map[k][j]<dis[j])
dis[j]=map[k][j];
}
for(int i=1;i<=n;i++) sum+=dis[i];
printf("%d\n",sum);
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
int w;
scanf("%d",&w);
map[i][j]=w;
}
prim();
}
}