朴素版的prim
作者:
巷港
,
2022-03-03 19:41:37
,
所有人可见
,
阅读 155
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N = 510;
const int INF = 0x3f3f3f3f;
int g[N][N]; //邻接矩阵表示图
int dist[N]; //表示当前点到已经联通的集合的最短距离
bool st[N];
int n,m;
int prim()
{
memset(dist,0x3f,sizeof dist);
int ans=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) ans+=dist[t]; //先累加!
st[t]=true;
for (int j=1;j<=n;j++)
dist[j]=min(dist[j],g[t][j]); //注意和迪杰斯特拉的区别!
}
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
memset(g,0x3f,sizeof g);
while (m--)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
g[a][b]=g[b][a]=min(g[a][b],c); //无向图是特殊的有向图
}
int ans=prim();
if (ans==INF) printf("impossible\n");
else printf("%d\n",ans);
return 0;
}