题目链接:http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66965#problem/L
题意:就是并查集题意。某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
以前用的是Prime算法,但是现在却几乎全部忘却了,由于记得并查集,所以,现在写Kruskal还是比较顺手的。
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct edge{
int u,v,w;
bool operator<(const edge &a)const{
return w<a.w;
}
}E[10010];
int N,M;
int pre[110];
int find(int x){
int t=x;
while(pre[t]!=t) t=pre[t];
while(x!=t) pre[x]=t,x=pre[x];
return t;
}
void Kruskal(){
for(int i=0;i<=N;i++) pre[i]=i;
int cnt=1,ans=0;
for(int i=0;i<M;i++){
int u=E[i].u,v=E[i].v,w=E[i].w;
int fu=find(u),fv=find(v);
if(fu==fv) continue;
ans+=w;
pre[fv]=fu;cnt++;
if(cnt==N) break;
}
cout<<ans<<endl;
}
int main(){
//freopen("D:\\in.txt","r",stdin);
while(cin>>N && N){
M=N*(N-1)/2;
for(int i=0;i<M;i++){
scanf("%d %d %d",&E[i].u,&E[i].v,&E[i].w);
}
sort(E,E+M);
Kruskal();
}
return 0;
}