HDU1879 继续畅通工程 (并查集)

继续畅通工程

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 18218 Accepted Submission(s): 7891

Problem Description

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

Input

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

当N为0时输入结束。

Output

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

Sample Input

Sample Output

3
1
0

Author

ZJU

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <ctime>
#include <cmath>
#include <string>
#include <cstring>
#include <stack>
#include <queue>
#include <list>
#include <vector>
#include <map>
#include <set>
using namespace std;

const int INF=0x3f3f3f3f;
const double eps=1e-10;
const double PI=acos(-1.0);
#define maxn 500
#define maxm 260000

struct Edge
{
    int u,v,w,s;
}edge[maxm];
int n,cnt1;
int ans;
int root[maxn];
int cmp(Edge a,Edge b)
{
    if(a.s != b.s)
        return a.s > b.s;
    return a.w < b.w;
}

void init()
{
    for(int i = 0; i < 101; i++)
        root[i] = i;
}
int find_root(int x)
{
    if(x != root[x])
        root[x] = find_root(root[x]);
    return root[x];
}
void uni(int a, int b, int c, int d)
{
    int x = find_root(a);
    int y = find_root(b);
    if(x != y)
    {
        root[y] = x;
        if(!d)
        ans += c;
        cnt1++;
    }
}
int main()
{
    while(~scanf("%d", &n) && n)
    {
        int m = n*(n-1)/2;
        for(int i = 0; i < m; i++)
            scanf("%d%d%d%d", &edge[i].u, &edge[i].v, &edge[i].w, &edge[i].s);
        init();
        sort(edge, edge+m, cmp);
        ans = 0;
        cnt1 = 0;
        for(int i = 0; i < m; i++)
        {
            uni(edge[i].u, edge[i].v, edge[i].w, edge[i].s);
            if(cnt1 == n-1)
                break;
        }
        printf("%d
", ans);
    }
    return 0;
}