最小生成树（Prim）

Prim算法思想:对图G(V,E)设置集合S，存放已被访问的顶点，然后每次从集合V-S中选择与集合S的最短距离最小的一个顶点(记为u),访问并加入集合S。之后，令顶点u为中介点，优化所有的从能到达的顶点v与集合S之间的最短距离，直到集合S已包含所有顶点。

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 impossible。

输入格式

第一行包含两个整数N,M，表示该图共有 N 个结点和 M 条无向边。

接下来 M 行每行包含三个整数 Xi​,Yi​,Zi​，表示有一条长度为 Zi​ 的无向边连接结点 Xi​,Yi​。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

输入输出样例

输入 

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

输出 

7

dis[i]是指i到最优集合的最小距离，而不是指i到某个点的最小距离。

st[i]数组用来标记点i是否加入到了集合。

我们规定先把点1加入集合， st[1]=true.

此时，不在集合中的点到集合的最小距离

dis[2]=min(dis[2],g[1][2])=2.

dis[3]=min(dis[3],g[1][3])=2.

dis[4]=min(dis[4],g[1][4])=3.

根据各个点到集合的最小距离，我们选择点2加入集合，st[2]=true. sum+=（点2到集合的距离=

dis[2]）

此时，不在集合中的点到集合的最小距离

dis[3]=min(dis[3],g[2][3])=2.

dis[4]=min(dis[4],g[2][4])=3.

根据各个点到集合的最小距离，我们选择点3加入集合，st[3]=true. sum+=（点3到集合的距离=

dis[3]）

此时，不在集合中的点到集合的最小距离

dis[4]=min(dis[4],g[3][4])=3.

把点4加入集合，st[4]=true.sum+=(点4到集合的距离=dis[4])

最后我们算出sum=7.

代码

//Prime法 求最小生成树
#include<bits/stdc++.h>
using namespace std;
int n, m;
const int N = 10000;
const int INF = 0x3f3f3f3f;
int dis[N], g[N][N];
bool st[N];

int prime(){
	memset(dis, INF, sizeof dis);
	int sum = 0;
	for(int i = 0; i < n; i++){
		int t = -1;
		for(int j = 1; j <= n; j++){
			if(!st[j] && (t == -1 || dis[j] < dis[t]))
				t = j;
		}
		st[t] = true;
		
		if(i && dis[t] == INF)return INF;
		if(i) sum += dis[t];
		for(int j = 1; j <= n; j++){
			dis[j] = min(dis[j], g[t][j]);
		}
	}
	return sum;
}
int main(){
	cin >> n >> m;
	memset(g, INF, sizeof g);
	for(int i = 1; i <= m; i++){
		int a, b, c;
		cin >> a >> b >> c;
		g[a][b] = g[b][a] = min(g[a][b], c);
	} 
	int t = prime();
	if(t == INF)cout << "impossible";
	else cout << t;
}