POJ 1258 Agri-Net（最小生成树-Prim）

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