Networking POJ - 1287 最小生成树水题 Prim 模板

题目链接http://poj.org/problem?id=1287

题意：

这道题其实就是在 N 个点的图中装宽带，要求全部互联。因为有重边，而且数据范围比较小，我们直接使用邻接矩阵存储，使用 Prim 算法就ok。

代码：

#include <iostream>
#include <cstdio>
#define N 55
#define E 2000
#define INF (1<<28)
#define fp(_p,_q,_r) for(int _p=_q;_p<_r;_p++)
using namespace std;
int n,e,cost[N][N],mincost[N];
bool used[N];
long long prim(){
    fp(i,0,N){
        mincost[i] = INF;
        used[i] = 0;
    }
    mincost[0] = 0;
    long long res = 0;
    while(true){
        int v = -1;
        fp(i,0,n){
            if(!used[i] && (v == -1 || mincost[i] < mincost[v])) v = i;
        }
        if(v == -1) break;
        used[v] = 1;
        res += mincost[v];
        fp(i,0,n){
            mincost[i] = min(mincost[i],cost[i][v]);
        }
    }
    return res;
}
int main(void)
{
    while(cin >> n && n && cin >> e){
        fill(cost[0],cost[0] + N * N,INF);
        int u,v,c;
        fp(i,0,e){
            cin >> u >> v >> c;
            u --,v --;
            cost[v][u] = cost[u][v] = min(cost[u][v],c);
        }
        cout << prim() << endl;
    }
    return 0;
}