Description
有n个农场,已知这n个农场都互相相通,有一定的距离,现在每个农场需要装光纤,问怎么安装光纤能将所有农场都连通起来,并且要使光纤距离最小,输出安装光纤的总距离
Input
第一行一个整数n表示农场个数(3<=n<=100),然后是n个农场的距离矩阵
Output
输出安装光纤的最小总距离
Sample Input
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
Sample Output
28
Solution
最小生成树,因为给出了一个二维矩阵代表他们的距离,直接算prim就行了
Code
#include<stdio.h>
#include<string.h>
#define INF 100000000
#define maxn 105
int n,sum;//n是顶点数,sum是结果
int map[maxn][maxn];//map表示距离矩阵
int used[maxn];//顶点i是否包含几何X中
int dis[maxn];//从集合X出发的边到每个顶点的最小权值
void Prim()
{
int i,j,k,min;
memset(used,0,sizeof(used));//初始化
used[1]=1;//从顶点1出发
for(i=1;i<=n;i++)//从顶点1出发的边到每个顶点的最小权值即顶点1到各顶点的距离
dis[i]=map[1][i];
for(i=1;i<n;i++)
{
min=INF;
for(j=1;j<=n;j++)//从不属于X的定点选取从X到其权值最小的顶点
{
if(!used[j]&&dis[j]<min)
{
min=dis[j];
k=j;
}
}
used[k]=1;//把定点k加到X中
sum+=min;//把边的长度加到结果中
for(j=1;j<=n;j++)
if(!used[j]&&dis[j]>map[k][j])
dis[j]=map[k][j];
}
}
int main()
{
int i, j;
while(scanf("%d",&n)!=EOF)
{
sum = 0;
for(i=1; i<=n; i++)
for(j=1; j<=n; j++)
scanf("%d", &map[i][j]);
Prim();
printf("%d\n",sum);
}
}