Prim算法求最小生成树

 

输入样例：

4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4

输出样例：

6

n*n=250000;m= 100000;则n*n与m相差不大，所以是稠密图，用邻接矩阵来存

#include<bits/stdc++.h>
using namespace std;
const int N = 550, inf = 0x3f3f3f3f;
int dis[N],g[N][N];
bool st[N];
int n, m, s, ed;

int prime() {
	memset(dis, 0x3f, sizeof dis);

	int res = 0;// 用来存放边之和
	dis[1] = 0;// 第一个点一定在最小生成树中
	for (int i = 0;i < n;i++) {
		int t = -1;

		for (int j = 1;j <= n;j++) {
			if (!st[j] && (t == -1 || dis[t] > dis[j])) {
				t = j;
			}
		}
		if ( dis[t] == inf) {//即有一个点根本到不了，所以需要跳出；
			return inf;
		}
		res += dis[t];
		st[t] = true;

		for(int j=1;j<=n;j++){
			dis[j] = min(dis[j], g[t][j]);//更新每个点到集合的距离；
		}
	}
	return res;
}


int main() {
	memset(g, 0x3f, sizeof g);

	cin >> n >> m;
	while (m--) {
		int a, b,c;
		cin >> a >> b>>c;
		g[a][b] = g[b][a] = min(g[a][b], c);
	}

	int ans = prime();

	if (ans == inf) {
		cout << "impossible" << endl;
	}
	else cout << ans << endl;
}