HDOJ1863 ( 畅通工程 ) 【最小生成树，kruscal】

根据《畅通工程》系列中的某题代码直接改的，忘记具体哪道了。

Problem : 1863 ( 畅通工程 )     Judge Status : Accepted
RunId : 5940637    Language : C++    Author : qq1203456195
Code Render Status : Rendered By HDOJ C++ Code Render Version 0.01 Beta

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 101
typedef struct{
    int x,y;
    int len;
}EDGE;
int set[N],n,r;
EDGE edge[N*N];
int cmp(const EDGE& a,EDGE& b)
{
    return a.len<b.len;
}
void MergeSet(int a,int b)
{
    int i;
    for (i=1;i<=n;i++)
        if(set[i]==a)
            set[i]=b;
}
int main()
{
    int i,p,sum,f,t;
    while (scanf("%d",&r),r)//n为村庄个数
    {
        scanf("%d",&n);
        //初始化set
        for (i=1;i<=n;i++)    set[i]=i;
        //读取边数
        for (i=0;i<r;i++)    scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].len);
        //已经选取的点的个数
        p=0;
        //路径总长度
        sum=0;
        //对边进行排序
        sort(edge,edge+i,cmp);
        for (i=0;i<r&&p!=n-1;i++)
        {
            //查找当前边的起始点是否在同一个集合中
            f=set[edge[i].x];
            t=set[edge[i].y];
            if(f!=t)
            {
                MergeSet(f,t);
                p++;
                sum+=edge[i].len;
            }
        }
        if(p!=n-1)
            printf("?\n");
        else
            printf("%d\n",sum);
    }
    return 0;
}