[kuangbin带你飞]专题六 最小生成树___poj1287(模板题)---kruskal

B - Networking POJ - 1287

思路：

有时模板题，练练手。

反思

1. 读边时，由于是生成树，所以不考虑方向。
2. 当节点数达到 n break（tot初始化为1，因为连通块里面本来就有一个了）

AC

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define mst(x,a) memset(x,a,sizeof(x))
#define For(i,x,y) for(register int i=(x); i<=(y); i++)
#define fzhead EDGE(int _bg, int _to, int _val, int _next)
#define fzbody bg(_bg), to(_to), val(_val), next(_next)
using namespace std;
typedef long long ll;
const int maxm=1e5+10;
const int maxn=100;
struct EDGE
{
    int bg,to,val,next;
    EDGE(){}
    fzhead:fzbody{}
}e[maxm];
int vis[maxn],f[maxn],cnt,head[maxn];
int n,m;
void add(int bg, int ed, int val)
{
    e[++cnt]=EDGE(bg,ed,val,head[bg]);
    head[bg]=cnt;
}
bool cmp(EDGE a, EDGE b)
{return a.val<b.val;}
int find(int x)
{
    if(x==f[x])return x;
    return f[x]=find(f[x]);
}
int main()
{
    while(scanf("%d",&n)&&n)
    {
        mst(head,-1);
        cnt=0;
        For(i,1,n)f[i]=i;
        scanf("%d", &m);
        int a,b,val,tot=1;
        For(i,1,m)
        {
            scanf("%d%d%d", &a, &b, &val);
            add(a,b,val);
        }
        sort(e+1,e+1+cnt,cmp);
        ll ans=0;
        For(i,1,cnt)
        {
            a=find(e[i].bg); b=find(e[i].to);
            if(a!=b)
            {
                f[a]=b;
                ans+=e[i].val;
                tot++;
            }
            if(tot==n)break;
        }
        printf("%lld\n",ans);
    }
    return 0;
}