题意: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;
}