首页 > 代码库 > POJ 1287 Networking

POJ 1287 Networking

最小生成树,也算是模版题。

依然Kruskal。N个点,M条边,排序,并查集。

没有什么坑点。


#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<iostream>
#include<list>
#include<set>
#include<cmath>
#define INF 0x7fffffff
#define eps 1e-6
using namespace std;
int fa[51],n,m;
int father(int x)
{
    if(x!=fa[x])
        fa[x]=father(fa[x]);
        return fa[x];
}
struct lx
{
    int u,v,len;
}l[10001];
bool cmp(lx a,lx b)
{
    return a.len<b.len;
}

int main()
{
    while(scanf("%d",&n),n)
    {
        for(int i=0;i<=n;i++)
            fa[i]=i;
        scanf("%d",&m);
        for(int i=0;i<m;i++)
            scanf("%d%d%d",&l[i].u,&l[i].v,&l[i].len);
        sort(l,l+m,cmp);
        int ans=0;
        for(int i=0;i<m;i++)
        {
            int fu=father(l[i].u);
            int fv=father(l[i].v);
            if(fu==fv)continue;
            fa[fv]=fu;
            ans+=l[i].len;
        }
        printf("%d\n",ans);
    }
}