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