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

来源：互联网发布：不出门便知天下事编辑：程序博客网时间：2024/06/05 11:23

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 141 2 42 3 83 4 74 5 95 6 106 7 27 8 18 9 72 8 113 9 27 9 63 6 44 6 141 8 8

Output示例

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

0 0