HDOJ1233 畅通工程之一（最小生成树Kruscal）

题目：

1233

还是畅通工程

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define M 101
typedef struct{
    int x,y;
    int len;
}EDGE;

EDGE edge[M*M];
int Map[M][M];
int n,m;
int set[M];

void ReadMap();
void InitSet();
int cmp(const EDGE &a,const EDGE &b);
int FindSet(int a);
void MergeSet(int a,int b);
void main()
{
    int c,s,f,t;
    while (scanf("%d",&n),n)
    {
        m = (n*(n-1))>>1;
        ReadMap();
        InitSet();
        c = 0;//choosed node
        s = 0;//sum length
        sort(edge,edge+m,cmp);//sort the edges
        for (int i=0;i<m && c!=n;i++)
        {
            f = FindSet(edge[i].x);
            t = FindSet(edge[i].y);
            if (f!=t)
            {
                MergeSet(f,t);
                c++;
                s += edge[i].len;
            }
        }
        cout<<s<<endl;

    }
}
void ReadMap()
{
    for (int i=0;i<m;i++)
        cin>>edge[i].x>>edge[i].y>>edge[i].len;
}
void InitSet()
{
    for (int i=1;i<=n;i++)
        set[i] = i;
}
int cmp(const EDGE &a,const EDGE &b)
{
    return a.len<b.len;
}
int FindSet(int a)
{
    return set[a];
}
void MergeSet(int a,int b)
{
    for (int i=1;i<=n;i++)
    {
        if (set[i] == a)
        {
            set[i] = b;
        }
    }
}