hdu - 1879 - 继续畅通工程

题意:N个村庄(N<100),给出每两个村庄之间建路的费用和是否建好的标记,求找这N个村庄连通的最少费用。roblem.p

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1879

——>>这是用来试试最小生成树的Kruskal与Prim算法的增强信心的好题呀……可惜第一次还是TLE,原因,用cin不用scanf,其中发现,用边数剪枝缩短时间不明显,还出现了这样剪枝用时更多的现象,最后,不剪枝了,直接用Kruskal。

#include <iostream>
#include <queue>
#include <cstdio>
#include <cstring>

using namespace std;

const int maxn = 100 + 10;      //N ( < 100 )
struct node     //结点数据类型
{
    int v1;     //其中一个村庄
    int v2;     //另一个村庄
    int cost;        //两个村庄之间的建路费用
    int ok;     //两个村庄是否已有路
    bool operator < (const node &v) const       //为优先队列重载运算符
    {
        return cost > v.cost;
    }
};
int fa[maxn], sum, N;      //并查集用的fa数组与最终目标sum

int find_set(int x)     //并查集查找
{
    if(x == fa[x]) return x;
    else
    {
        fa[x] = find_set(fa[x]);
        return fa[x];
    }
}
void union_set(node n)      //并查集合并
{
    int root_v1 = find_set(n.v1);
    int root_v2 = find_set(n.v2);
    if(root_v1 == root_v2) return;
    else
    {
        fa[root_v2] = root_v1;
        if(!n.ok) sum += n.cost;        //没建有路时才加上费用
        return;
    }
}
int main()
{
    int i;
    node no;
    while(cin>>N)
    {
        if(!N) return 0;
        for(i = 1; i <= N; i++) fa[i] = i;      //初始化
        priority_queue<node> pq;        //为Kruskal算法用的优先队列
        int bor = N*(N-1)/2;
        for(i = 0; i < bor; i++)
        {
            scanf("\n%d%d%d%d", &no.v1, &no.v2, &no.cost, &no.ok);
            if(!no.ok) pq.push(no);
            else union_set(no);
        }
        sum = 0;
        while(!pq.empty())
        {
            no = pq.top();
            pq.pop();
            union_set(no);
        }
        printf("%d\n", sum);
    }
    return 0;
}



你可能感兴趣的:(hdu - 1879 - 继续畅通工程)