九度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<iostream>
#include<algorithm>

using namespace std;
struct R {
       int a, b;
       int len;
};
R r[10002];
int root[1002];

int findroot(int x) {
    if (root[x] == -1)
       return x;
    else {
         int t = findroot(root[x]);
         root[x] = t;
         return t;
         }
}

bool comp(const R &r1, const R &r2) {
     return r1.len < r2.len;
}

int main()
{
    int n, m;
    while ( scanf("%d %d", &n, &m) != EOF) {
          for (int i=1; i<=n; ++i)
              root[i] = -1;
          for (int i=0; i!=m; ++i)
              scanf("%d%d%d" ,&r[i].a ,&r[i].b, &r[i].len);
       int sum = 0, count = 0;
       sort (r, r+m, comp);
       for (int i=0; i!=m; ++i) {
           int a = findroot(r[i].a);
           int b = findroot(r[i].b);
           if (a != b) {
             root[a] = b;
              ++count;
              sum += r[i].len; 
              if (count == n-1) //全部联通 
                break;
           }
       }
       if (count != n-1)
         printf("no\n");
       else
        printf("%d\n", sum);
    }
    return 0;
}