题目1028:继续畅通工程

#include <stdio.h>
#include <algorithm>
using namespace std;
#define N 101
 
int tree[N];
int find(int x)
{
    if (tree[x] == -1) return x;
    else
    {
        int tmp = find(tree[x]);
        tree[x] = tmp;
        return tmp;
    }
}
 
struct Edge
{
    int a, b;
    int cost;
    int is;
    bool operator < (const Edge &A) const
    {
        return cost < A.cost;
    }
};
 
Edge edge[5000];
 
int main()
{
    int n;
    while (scanf("%d", &n) != EOF&&n != 0)
    {
        int size = 0;
        for (int i = 1; i <= n; i++)
            tree[i] = -1;
        for (int i = 1; i <= n*(n - 1) / 2; i++)
        {
            int a, b, cost, is;
            scanf("%d %d %d %d", &a, &b, &cost,&is);
            if (is == 1)
            {
                int x = find(a);
                int y = find(b);
                if (x != y)
                    tree[x] = y;
            }
            else
            {
                edge[size].a = a;
                edge[size].b = b;
                edge[size].cost = cost;
                size++;
            }
        }
        sort(edge, edge + size-1);
        int ans = 0;
        for (int i = 0; i < size; i++)
        {
            int a = find(edge[i].a);
            int b = find(edge[i].b);
            if (a != b)
            {
                tree[a] = b;
                ans += edge[i].cost;
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}
/**************************************************************
    Problem: 1028
    User: cust123
    Language: C++
    Result: Accepted
    Time:10 ms
    Memory:1100 kb
****************************************************************/


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值