hdu1233

http://acm.hdu.edu.cn/showproblem.php?pid=1233

最小生成树，kruskal算法

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
const int N=5005;
struct stu{
    int u;
    int v;
    int w;
}p[N];
int n,m;
int father[N];

int cmp(const void *a,const void *b)
{
    return (*(struct stu*)a).w > (*(struct stu*)b).w?1:-1;
}
int find(int x)
{
    if(father[x]!=x)
    father[x]=find(father[x]);
    return father[x];
}
int make(int a,int b)
{
    int f1=find(a);
    int f2=find(b);
    if(f1!=f2)
    {
        father[f2]=f1;
        return 1;
    }
    return 0;
}
int kruskal()
{
    int cns=0,k=0;
    for(int i=0;i<m;i++)
    {
        if(make(p[i].u,p[i].v))
        {
            cns+=p[i].w;
            k++;
        }
        if(k==n-1)
        return cns;
    }
    return cns;
}

int main()
{
    while(~scanf("%d",&n))
    {
        if(!n)
        break;
        m=n*(n-1)/2;
        for(int i=1;i<=n;i++)
        father[i]=i;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].w);
        }
        qsort(p,m,sizeof(struct stu),cmp);
        printf("%d
",kruskal());

    }
    return 0;
}