POJ 1258 Agri-Net

最小生成树prim

原理：

　　任取一点作为起始点，遍历当前点的所有的邻近点，维护更新到达邻近点的最小花费。选择权值最小的点作为下一次的起始点；

　　已经遍历过的点不再作为起始点。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;

const int INF = 99999999 ;
const int maxn = 105 ;

int map[maxn][maxn];
int visit[maxn];
int low[maxn];

int main (){
    int n;
    while (~scanf ("%d",&n)){
        for (int i=1;i<=n;i++){
            for (int j=1;j<=n;j++){
                scanf ("%d",&map[i][j]);
            }
            map[i][i]=INF;
        }
        int ans=0;
        memset (visit,0,sizeof visit);
        int pos=1;visit[1]=1;
        for (int i=1;i<=n;i++)
            low[i]=map[pos][i];
        int mi=INF;
        for (int i=1;i<n;i++){
            mi=INF;
            for (int j=1;j<=n;j++)
                if (visit[j]==0&&low[j]<mi){
                    mi=low[j];pos=j;
                }
            ans+=mi;visit[pos]=1;
            for (int j=1;j<=n;j++)
                if (!visit[j])
                    low[j]=min(low[j],map[pos][j]);
        }
        printf ("%d
",ans);
    }
    return 0;
}