九度oj 题目1028：继续畅通工程

题目描述：

    省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

输入：

    测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

    当N为0时输入结束。

输出：

    每个测试用例的输出占一行，输出全省畅通需要的最低成本。

样例输入：

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

样例输出：

3
1
0

#include<iostream>
#include<algorithm>

using namespace std;

struct E {  //边 
       int a;
       int b;
       int cost;
       bool isbuild;
};

E edge[6000];
int root[100]; //树根 

int findroot(int x) {  //找该树的根 
    if (root[x] == -1)
      return x;
    else {
         int t = findroot(root[x]);
         root[x] = t;
         return t;
    }
}

bool  comp(E e1, E e2) {
      return e1.cost < e2.cost;
}

int main()
{
    int n;
    while (cin >> n && n) {
       for (int i=1; i<=n; ++i)
              root[i] = -1;
          int m = n*(n-1)/2;
       for (int i=1; i<=m; ++i) {
         cin >> edge[i].a >> edge[i].b 
             >> edge[i].cost >> edge[i].isbuild;
         if (edge[i].isbuild) //已建的花费为0 
           edge[i].cost = 0;
       }
    sort (edge+1, edge+m+1, comp); //注意从1开始 
    int sum = 0, count = 0;
    for (int i=1; i<=m; ++i) { 
           int a = findroot( edge[i].a);
           int b = findroot (edge[i].b);
           if (a != b) { //不同并查集合并 
              root[a] = b;
              ++count;    
              sum += edge[i].cost;
              if (count == n-1)  //全部联通 
                break;
            }
    }
    cout << sum << endl;
  }//while
  return 0;
}