poj 1287 Networking

点击打开链接poj 287

思路：最小生成树 + prime

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAXN 60
#define INF 0xFFFFFFF

int n , m;
int vis[MAXN];
int lowcost[MAXN];
int G[MAXN][MAXN];

void init(){
    memset(vis , 0 , sizeof(vis));
    memset(lowcost , 0 , sizeof(lowcost));   
    for(int i = 1 ; i <= n ; i++){
       for(int j = 1 ; j <= n ; j++)
           G[i][j] = INF;
    }
}

void prime(){
     int ans , pos;
     for(int i = 1 ; i <= n ; i++)
         lowcost[i] = G[1][i];
     ans = 0;
     vis[1] = 1;
     for(int i = 1 ; i <= n ; i++){
        pos = -1;
        for(int j = 1 ; j <= n ; j++){
           if(!vis[j] && (pos == -1 || lowcost[j] < lowcost[pos]))
             pos = j;
        }
        if(pos == -1)
            break;
        vis[pos] = 1;
        ans += lowcost[pos];
        for(int j = 1 ; j <= n ; j++){
           if(!vis[j] && lowcost[j] > G[j][pos])
             lowcost[j] = G[j][pos];
        }
     }
     printf("%d\n" , ans);
}

int main(){
    int x , y , v;
    while(scanf("%d%d" , &n , &m) && n){
         init();
         if(!m)
             printf("0\n");
         else{ 
             for(int i = 0 ; i < m ; i++){
                scanf("%d%d%d" , &x , &y , &v);
                if(G[x][y] > v)
                   G[x][y] = G[y][x] = v;
             }
             prime();
         }
    }
}