九度oj 题目1347：孤岛连通工程

题目描述：

现在有孤岛n个，孤岛从1开始标序一直到n，有道路m条（道路是双向的，如果有多条道路连通岛屿i，j则选择最短的那条），请你求出能够让所有孤岛都连通的最小道路总长度。

输入：

数据有多组输入。
每组第一行输入n(1<=n<=1000),m(0<=m<=10000)。
接着m行，每行输入一条道路i j d(0<=d<=1000)，（i,j表示岛屿序号，d表示道路长度）。

输出：

对每组输入输出一行，如果能连通，输出能连通所有岛屿的最小道路长度，否则请输出字符串"no"。

样例输入：

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

样例输出：

3
no

求最小生成树

 #include <cstdio>
 #include <algorithm>
 #define max_v 1002
 #define max_m 10002
 struct Edge{
     int from, to, dis;
 };
 int par[max_v];
 int rank[max_v];

 int cmp(const void *a, const void *b) {
     Edge at = *(Edge *)a;
     Edge bt = *(Edge *)b;
     return at.dis - bt.dis;
 }
 int ini(int N) {
     for(int i = ; i <= N; i++) {
         par[i] = i;
         rank[i] = ;
     }
 }
 int find(int t) {
     if(t == par[t]) {
         return t;
     }
     else {
         return par[t] = find(par[t]);
     }
 }

 Edge e[max_m];
 int n, m;
 int main(int argc, char const *argv[])
 {
     freopen("input.txt","r",stdin);
     while(scanf("%d %d",&n,&m) != EOF) {
         for(int i = ; i < m; i++) {
             scanf("%d %d %d",&e[i].from, &e[i].to, &e[i].dis);
         }
         qsort(e, m, sizeof(Edge), cmp);
         int ans = ;
         int cnt = ;
         ini(n);
         for(int i = ; i < m; i++) {
             int a = e[i].from;
             int b = e[i].to;
             int c = e[i].dis;
             //printf("%d %d %d\n",a,b,c);
             a = find(a);
             b = find(b);
             if(a == b) {
                 continue;
             }
             if(a != b) {
                 ans = ans + c;
                 if(rank[a] < rank[b]) {
                     par[a] = b;
                 }
                 else {
                     par[b] = a;
                     rank[a]++;
                 }
                 cnt++;
                 if(cnt == n-) {
                     break;
                 }
             }
         }
         if(cnt == n-) {
             printf("%d\n",ans);
         }
         else {
             puts("no");
         }
     }
     return ;
 }