问题
假设给定一个加权连通图G,G的边集合为E,顶点个数为n,要求其一棵最小生成树T。
思路
将图中所有的边按照边权从小到大排序。遍历排序好的边集,若该条边的左右端点不在同一个集合内,则将该条边加入最小生成树。重复上述过程,直至最小生成树中有n-1条边为止或遍历完所有边(无解)为止。
代码
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
struct asdf
{
int x,y,z;
}a[200001];
int f[5001],n,m;
bool cmp(asdf a,asdf b)
{
return a.z<b.z;
}
int findd(int x)
{
return x==f[x]?x:f[x]=findd(f[x]);
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;++i) f[i]=i;
for (int i=1,t;i<=m;++i) scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
sort(a+1,a+1+m,cmp);
long long ans=0;
for (int i=1;i<=m;++i)
if (findd(a[i].x)!=findd(a[i].y))
f[f[a[i].x]]=f[a[i].y],ans+=a[i].z;
for (int i=2;i<=n;++i)
if (f[i]!=f[1])
{
printf("orz\n");
return 0;
}
printf("%lld\n",ans);
}