最小生成树问题
1.朴素版Prim(权重正负都实用)
举例模拟一遍:
4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4
注意:与Dijkstra算法不同的是Dijkstra算法更新的是每个点到第一个点的最短距离,而Prim算法更新的是到生成树的最短距离。
#include<iostream>
#include<cstring>
using namespace std;
const int N = 510 , INF = 0x3f3f3f3f;
int g[N][N];
int n,m;
int dist[N];//存放更新的边长
bool st[N];//判断是否已经确定边长
int Prim()
{
int res = 0;
for(int i=0;i<n;i++)
{
int t = -1;
for(int j=1;j<=n;j++)
if(!st[j] && (t == -1 || dist[t] > dist[j])) t = j;//找到每次更新后最小的边
if(i && dist[t] == INF) return INF;//说明此图不连通
if(i) res += dist[t];//更新权重
for(int j=1;j<=n;j++)
dist[j] = min(dist[j] , g[t][j]);//用最小的点更新该点能到达的所有距离。
st[t] = true;//标记t号点已经确定。
}
return res;
}
int main()
{
cin>>n>>m;
memset(dist,0x3f,sizeof dist);
memset(g,0x3f,sizeof g);
for(int i=0;i<m;i++)
{
int a,b,c;
cin>>a>>b>>c;
g[a][b] = g[b][a] = min(g[a][b] , c);
}
int t = Prim();
if(t == INF) cout<<"impossible";
else cout<<t;
}