HDU 1233还是畅通工程(简单最小生成树)
问题描述:
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
样例输入:
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
样例输出:
对每个测试用例,在1行里输出最小的公路总长度。
3
5
刚学习完并查集合最小生成树的相关算法,所以采用kruskal算法构建最小生成树,通过最小堆存储所有边的权值,依次取出当前权值最小的边,当该条边不属于已经有的连通分支的时候,将该边加入已有的连同分支。
AC代码:
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;
typedef struct Edge
{
int a, b;
int weight;
bool operator < (Edge e) const
{
return weight < e.weight;
}
bool operator > (Edge e) const
{
return weight > e.weight;
}
}Edge;
int pre[105];
int Find(int x)
{
int r = x;
while(pre[r] != r)
{
r = pre[r];
}
int i = x, j;
while(i != r)
{
j = pre[i];
pre[i] = r;
i = j;
}
return r;
}
void join(int x, int y)
{
int fx = Find(x), fy = Find(y);
if(fx != fy)
{
pre[fx] = fy;
}
}
int main()
{
int n;
while(~scanf("%d", &n) && n != 0)
{
Edge e[n * (n - 1) / 2];
int x, y, w;
priority_queue<Edge, vector<Edge>, greater<Edge> > q;
for(int i = 0; i < n * (n - 1) / 2; i++)
{
scanf("%d %d %d", &x, &y, &w);
e[i].a = x;
e[i].b = y;
e[i].weight = w;
q.push(e[i]);
pre[x] = x;
pre[y] = y;
}
int sum = 0;
for(int i = 0; i < n * (n - 1) / 2; i++)
{
if(Find(q.top().a) != Find(q.top().b))
{
join(q.top().a, q.top().b);
sum += q.top().weight;
q.pop();
}
else
{
q.pop();
}
}
printf("%d\n", sum);
}
return 0;
}