HDU 1233 还是畅通工程

#include <iostream>
#include <cstdio>
#include <algorithm>
const int MAXN=11000;
using namespace std;
int u[MAXN],v[MAXN],w[MAXN],p[MAXN],r[MAXN],n,m;
int cmp(const int i,const int j){return w[i]<w[j];}//紫书上教的间接排序，好像挺好用的样子
int find(int x){return p[x]==x?x:p[x]=find(p[x]);}
int Kruskal(){
    int ans=0;
    for(int i=1;i<=n;i++)p[i]=i;
    for(int i=1;i<=m;i++)r[i]=i;
    sort(r+1,r+1+m,cmp);
    for(int i=1;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()
{
    while(scanf("%d",&n)&&n)
    {
        m=n*(n-1)/2;
        for(int i=1;i<=m;i++)
        scanf("%d%d%d",&u[i],&v[i],&w[i]);
        printf("%d\n",Kruskal());
    }
}