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