Prim模板

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Fantasy_World/article/details/82935604

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define debug(x) cerr << #x << "=" << x << endl;
const int MAXN = 5010; 
int n,m,gra[MAXN][MAXN],ans,d[MAXN],vis[MAXN];
void prim() {
	memset(d, 0x3f, sizeof(d));
	memset(vis, 0, sizeof(vis));
	d[1] = 0;
	for(int i=1; i<=n; i++) {
		int x = 0;
		for(int j=1; j<=n; j++) {
			if(!vis[j] && ( x == 0 || d[j] < d[x] )) x = j;//注意这里的括号，&&右边的两个条件要括起来
		}
		vis[x] = 1;
		for(int j=1; j<=n; j++) {
			if(!vis[j]) d[j] = min(d[j], gra[x][j]);
		}
	}
}
int main() {
	scanf("%d%d", &n, &m);
	memset(gra, 0x3f, sizeof(gra));
	for(int i=1; i<=n; i++) gra[i][i] = 0;//这里的初始化。。。
	for(int i=1; i<=m; i++) {
		int x,y,z;
		scanf("%d%d%d", &x, &y, &z);
		gra[x][y] = gra[y][x] = min(gra[x][y], z);
	}
	prim();
	for(int i=2; i<=n; i++) {
		ans += d[i];
	}
	printf("%d", ans);
	return 0;
}