HDU 1233 还是畅通工程(Kruskal算法)

时间:2023-03-09 03:13:38
HDU 1233 还是畅通工程(Kruskal算法)

题目链接:

http://acm.hdu.edu.cn/showproblem.php?pid=1233

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 56123    Accepted Submission(s): 25465

Problem Description
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省*“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最小的公路总长度。
Sample Input
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
Sample Output
3
5
Hint

Hint

Huge input, scanf is recommended.

分析:
 裸Kruskal算法,注意给你边的个数不是n个,而是n*(n-1)/2个
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define max_v 10005
struct edge
{
int x,y;
int w;
};
edge e[max_v];
int rk[max_v];
int pa[max_v];
int sum;
bool cmp(edge a,edge b)
{
return a.w<b.w;
}
void make_set(int x)
{
pa[x]=x;
rk[x]=0;
}
int find_set(int x)
{
if(x!=pa[x])
pa[x]=find_set(pa[x]);
return pa[x];
}
void union_set(int x,int y,int w)
{
x=find_set(x);
y=find_set(y);
if(x==y)
return ;
if(rk[x]>rk[y])
{
pa[y]=x;
}else
{
if(rk[x]==rk[y])
rk[y]++;
pa[x]=y;
}
sum+=w;
return ;
}
int main()
{
int n;
while(~scanf("%d",&n))
{
sum=0;
if(n==0)
break;
for(int i=1;i<=n;i++)
make_set(i);
n=(n-1)*n/2.0;//题目要求的,而不是算法要求的
for(int i=0;i<n;i++)
{
scanf("%d %d %d",&e[i].x,&e[i].y,&e[i].w);
}
sort(e,e+n,cmp);
for(int i=0;i<n;i++)
{
union_set(e[i].x,e[i].y,e[i].w);
}
printf("%d\n",sum);
}
}
/*按照边的权重排序,每次选择max/min 选择某编的时候如果构成了环,就不选
//解决:加权无向图