Prim模板

大體過程
挑一個未處理集合中“邊值”最小的一個點
把這個點放入已處理集合
用這個點所連的邊更新一波“邊值”

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