还是畅通工程（最小生成树，并查集）

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

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <vector>
#include <algorithm>
using namespace std;
struct Edge
{
    int a, b;//两个顶点编号
    int cost;//边的权值
} edge[6000];
int tree[200];
int findroot(int x){//找根
    if(tree[x] == -1) return x;
    else return findroot(tree[x]);
}
int cmp(Edge x, Edge y){
    return x.cost < y.cost;
}
int main(int argc, char** argv)
{
    int n;
    while(cin >> n)
    {
        if(n == 0) break;
        memset(tree, -1, sizeof(tree));
        for(int i = 1; i <= n * (n- 1) /2; i++)
        {
            cin >> edge[i].a >> edge[i].b >> edge[i].cost;
        }
        sort(edge + 1, edge + 1 + n * (n - 1) / 2, cmp);//按权值从小到大排
        int ans = 0;//权值和
        for(int i = 1; i <= n * (n - 1)/2; i++)//按边权值递增顺序遍历所有边
        {
            int a = findroot(edge[i].a);
            int b = findroot(edge[i].b);
            if(a != b)
            {
                tree[a] = b;//合并两个集合
                ans += edge[i].cost;//累加权值
            }
        }
        cout << ans << endl;
    }

    return 0;
}