题目1028：继续畅通工程

题目描述：

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

输入：

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

    当N为0时输入结束。

输出：

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

样例输入：

31 2 1 01 3 2 02 3 4 031 2 1 01 3 2 02 3 4 131 2 1 01 3 2 12 3 4 10

样例输出：

310

代码：

#include <stdio.h>#include <algorithm>using namespace std;int tree[100];int findRoot(int x) {    if(tree[x] == -1) return x;    else {int tmp = findRoot(tree[x]);tree[x] = tmp;return tmp;    }}struct Edge {    int s;    int e;    int d;}edge[5000];struct Edged {    int s;    int e;    int d;}edged[5000];bool cmp1(Edge A,Edge B) {    return A.d < B.d;}bool cmp2(Edged A,Edged B) {    return A.d < B.d;}int main() {    int n;    while(scanf("%d",&n)!=EOF) {if(n == 0) break;int size1 = 0,size2 = 0;int x,y,z,build;for(int i=0;i<n*(n-1)/2;i++) {    scanf("%d %d %d %d",&x,&y,&z,&build);    if(build) {edged[size1].s = x;edged[size1].e = y;edged[size1].d = z;size1++;    }    else {edge[size2].s = x;edge[size2].e = y;edge[size2].d = z;size2++;    }}for(int i=1;i<=n;i++)    tree[i] = -1;sort(edge,edge+size2,cmp1);sort(edged,edged+size1,cmp2);for(int i=0;i<size1;i++) {    int a = findRoot(edged[i].s);    int b = findRoot(edged[i].e);    if(a != b)tree[a] = b;}int sum = 0;for(int i=0;i<size2;i++) {    int a = findRoot(edge[i].s);    int b = findRoot(edge[i].e);    if(a != b) {tree[a] = b;sum += edge[i].d;    }}printf("%d\n",sum);    }    return 0;}