poj 1258 Agri-Net 基本&经典最小生成树

传送门：poj 1258 Agri-Net

题目大意

n个农场，已知这n个农场都互相相通，有一定的距离，现在每个农场需要装光纤，问怎么安装光纤能将所有农场都连通起来，并且要使光纤距离最小，输入的是邻接矩阵，输出安装光纤的总距离

解题思路

裸prim，
1. 随便找一个元素进行寻找
2. 找出和这个元素相邻的所有权值，如果不能到达表示无限大
3. 找到和这个元素相邻的最小权值，记录下来
4. 最小权值累加并且标记那个位置
5. 把位置转换为和上一个元素相邻的最小权值的位置，病更新权值

AC代码

#include<cstdio>#include<cstring>const int INF = 0x3f3f3f3f;const int MAXN = 120;int map[MAXN][MAXN],low[MAXN],vis[MAXN];int n;int prim(){    int pos,ret = 0,Min;    memset(vis,0,sizeof vis);    pos = 1,vis[1] = 1;    for(int i=1;i<=n;i++)        if(i!=pos) low[i] = map[pos][i];    for(int i=1;i<n;i++)    {        Min = INF;        for(int j=1;j<=n;j++)            if(!vis[j] && low[j]<Min ) Min = low[j],pos = j;        ret += Min,vis[pos] = 1;        for(int j=1;j<=n;j++)            if(!vis[j] && low[j]>map[pos][j] ) low[j] = map[pos][j];    }    return ret;}int main(){    int v;    while(~scanf("%d",&n))    {        for(int i=1;i<=n;i++)            for(int j=1;j<=n;j++)            {                scanf("%d",&v);                map[i][j] = map[j][i] = v;            }        printf("%d\n",prim());    }    return 0;}