51nod 1212 无向图最小生成树（prim）

1212 无向图最小生成树

基准时间限制：1 秒 空间限制：131072 KB 分值: 0  难度：基础题

 收藏

 关注

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

//prim算法，与最短路径 Dijkstra相似 
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define MAX 1005
#define inf 0x3f3f3f3f
int mat[MAX][MAX];
int dist[MAX],visit[MAX];

int prim(int n){
	int i,j;
	int sum=0;
	memset(visit,0,sizeof(visit));
	visit[1]=1;
	for( i=2;i<=n;i++){
		dist[i]=mat[1][i];
	}
	dist[1]=0;
	for( i=2;i<=n;i++){//一定是循环n-1次
		int minn=inf,k;
		for(j=1;j<=n;j++){
			if(!visit[j]&&dist[j]<minn){
				minn=dist[j];
				k=j;
			}
		} 
			visit[k]=1;
			sum+=minn;
			if(inf==minn)
			return -1;
			for(int j=1;j<=n;j++){
				if(!visit[j]&&dist[j]>=mat[k][j])
				dist[j]=mat[k][j];
			}
		}
	return sum;
}
int main()
{
	int a,b,c;
	int n,m;
	scanf("%d%d",&n,&m);
	memset(mat,inf,sizeof(mat));
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&a,&b,&c);
		if(mat[a][b]>c){
			mat[a][b]=c;
			mat[b][a]=c;
		}
	}
	printf("%d\n",prim(n));
	return 0;
}