51nod 1212 无向图最小生成树prim算法

N个点M条边的无向连通图，每条边有一个权值，求该图的最小生成树。

Input

第1行：2个数N,M中间用空格分隔，N为点的数量，M为边的数量。（2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行：每行3个数S E W，分别表示M条边的2个顶点及权值。(1 <= S, E <= N，1 <= W <= 10000)

Output

输出最小生成树的所有边的权值之和。

Input示例

9 14
1 2 4
2 3 8
3 4 7
4 5 9
5 6 10
6 7 2
7 8 1
8 9 7
2 8 11
3 9 2
7 9 6
3 6 4
4 6 14
1 8 8

Output示例

37

#include <iostream>
#include <stdio.h>
#include <string.h>
const int INF = 0x3f3f3f3f;
using namespace std;
int map[1010][1010];
int dist[1010];
int vis[1010];
int n,m,x,y,z;
void Init(){
    for(int i = 0;i <= n;i ++)
        for(int j = 0;j <= n;j ++)
            map[i][j] = map[j][i] = INF;
}
int prim(int v0){
    int i,j,sum = 0;
    for(int i = 1;i <= n;i ++){
        dist[i] = map[v0][i];
        vis[i] = v0;
    }
    vis[v0] = -1;
    for(int i = 1;i < n;i ++){
        int min = INF,v = -1;
        for(int j = 1;j <= n;j ++){
            if(vis[j] != -1 && dist[j] < min){
                v = j;
                min = dist[j];
            }
        }
        if(v != -1){
            vis[v] = -1;
            sum += dist[v];
            for(int j = 1;j <= n;j ++){
                if(vis[j] != -1 && map[v][j] < dist[j]){
                    dist[j] = map[v][j];
                    vis[j] = v;
                }
            }
        }
    }
    return sum;
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        Init();
        for(int i = 0;i < m;i ++){
            scanf("%d%d%d",&x,&y,&z);
            if(map[x][y] > z)
                map[x][y] = map[y][x] = z;
        }
        printf("%d\n",prim(1));
    }
    return 0;
}