HDU1233还是畅通工程(最小生成树 Kruskal算法)

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

//结构体做法

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
int p[101],sum,num;    //p为并查集
struct edge//图的结构体
{
    int sv,ev,w;//起始边，终边，权值。
};

struct edge e[10000];//必须开到大于(n*n-1)/2，否则必然RE。。。

int comp(const void *a,const void *b){
    return (*(struct edge*)a).w > (*(struct edge*)b).w ?1:-1;
}

int find(int x){//并查集find函数
    return p[x] == x ? x : (p[x] = find(p[x]));//压缩路径
}

void merge(int x,int y,int w){//并查集merge函数
    x = find(x); y = find(y);
    if(x != y){
        p[x] = y; sum += w; num++;
    }
}

int main(){
    int n,m,i;
    while(scanf("%d",&n)==1 && n != 0)
    {
        sum = 0; num = 1;
        m = n*(n-1)/2;
        for(i = 1;i <= n;i ++) p[i] = i;//初始化并查集
        for(i = 0;i < m;i ++){
            scanf("%d %d %d",&e[i].sv,&e[i].ev,&e[i].w);
        }
        qsort(e,m,sizeof(e[0]),comp);//调用结构体快排
        for(i = 0; i < m; i++){
            merge(e[i].sv,e[i].ev,e[i].w;
            if(num == n) break;
        }
        printf("%d\n",sum);
    }
    return 0;
}

//直接数组

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define MAXN 6000

int u[MAXN], v[MAXN], w[MAXN], r[MAXN], p[MAXN];

int cmp(const void *a, const void *b){
    int x = *(int *)a, y = *(int *)b;
    return w[x] - w[y];
}

int find(int x){return p[x] == x ? x : (p[x] = find(p[x]));}

int Kruskal(int n, int m){
    int ans = 0, i;
    for(i=0; i<n; i++) p[i] = i;//初始化并查集
    for(i=0; i<m; i++) r[i] = i;//初始化边号
    qsort(r, m, sizeof(r[0]), cmp);
    for(i=0; i<m; i++){
        int e = r[i]; int x = find(u[e]); int y = find(v[e]);
        if(x != y) {ans += w[e]; p[x] = y;}
    }
    return ans;
}

int main(){
    int n, m, i;
    while(scanf("%d", &n) == 1 && n != 0){
        m = n*(n-1)/2;
        for(i=0; i<m; i++){
            scanf("%d %d %d", &u[i], &v[i], &w[i]);
            u[i]--; v[i]--;//从零开始编号
        }
        printf("%d\n", Kruskal(n, m));
    }
    return 0;
}

//如果是从1开始编号 只需要注释掉
//u[i]--; v[i]--;
//并且把for(i=0; i<n; i++) p[i] = i;//初始化并查集
//改成for(i=1; i<=n; i++) p[i] = i;//初始化并查集