HDOJ1233-----最小生成樹簡單題

Problem Description 某省調查鄉村交通狀況，得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目标是使全省任何兩個村莊間都可以實作公路交通（但不一定有直接的公路相連，隻要能間接通過公路可達即可），并要求鋪設的公路總長度為最小。請計算最小的公路總長度。

Input 測試輸入包含若幹測試用例。每個測試用例的第1行給出村莊數目N ( < 100 )；随後的N(N-1)/2行對應村莊間的距離，每行給出一對正整數，分别是兩個村莊的編号，以及此兩村莊間的距離。為簡單起見，村莊從1到N編号。

當N為0時，輸入結束，該用例不被處理。

Output 對每個測試用例，在1行裡輸出最小的公路總長度。

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
        

Sample Output

3
5
       

#include<cstdio>  
#include<algorithm>  
using namespace std;   
int father[102];  
struct node{  
    int a, b, value;  
}w[5002];  
bool cmp(const node &a,const node &b){  
    return a.value < b.value;  
}  
int find(int x){  
    return father[x] == x ? x : (father[x] = find(father[x]));//壓縮路徑
}  
void merge(int x, int y, int z, int *tp){  
    int a = find(x), b = find(y);  
    if(a != b){  
        (*tp) += z;  
        father[a] = b;  
    }  
}  
int main()  
{  
    int n, m, sum;  
    while(scanf("%d", &n) && n)  
    {  
        sum = 0;  
        m = n * (n-1) / 2;  
        for(int i = 1; i <= m; i++){
        	scanf("%d%d%d",&w[i].a, &w[i].b, &w[i].value);
		}  
        sort(w+1, w+m+1, cmp);//按邊長從小到大排序  
        for(int i = 1; i <= n; i++){
        	father[i] = i;
		}  
        for (int i = 1; i <= m; i++)  {  
            merge(w[i].a, w[i].b, w[i].value, &sum);  
        }  
        printf("%d\n",sum);  
    }  
}