HDU 1233 还是畅通工程

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

题意：

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

思路：

我用了Kruskal算法。

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

const int maxn = 5000 + 5;

int n, ans;
int p[maxn];

struct node
{
int u;
int v;
int w;
}edge[maxn];

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

bool cmp(node a, node b)
{
return a.w < b.w;
}

void Kruskal()
{
int cnt = n*(n - 1) / 2;
sort(edge, edge + cnt, cmp);
for (int i = 0; i < cnt; i++)
{
int x = find(edge[i].u);
int y = find(edge[i].v);
if (x != y)
{
p[x] = y;
ans += edge[i].w;
}
}
}

int main()
{
//freopen("D:\\txt.txt", "r", stdin);
while (scanf("%d",&n) && n)
{
for (int i = 1; i <= n; i++)
p[i] = i;
for (int i = 0; i < n*(n - 1) / 2; i++)
{
scanf("%d%d%d", &edge[i].u, &edge[i].v , &edge[i].w);
}
ans = 0;
Kruskal();
printf("%d\n",ans);
}
return 0;
}