用Kruskal算法解POJ 1251 Jungle Roads代码

#include<stdio.h>
#include<algorithm>
using namespace std;

struct road{
int x,y,price;   //price表示连接 x，y村庄的公路维修费用
};
int cmp(road a,road b){
return a.price<b.price;
}
int deal(int n){
int i,j,k=0,q,p,root[27],js=1,total=0;    //k--总边数  q--每一行边数  p--费用  root[]表示根 total--最少费用
road s[80];
char a,b;    //a--x， b--y

for(i=0;i<n-1;i++){

scanf("\n%c %d ",&a,&q);
for(j=0;j<q;j++){
scanf("%c %d ",&b,&p);
s[k].x=a-'A', s[k].y=b-'A', s[k++].price=p;
//root[b-'A']=a-'A';
}
}
sort(s,s+k,cmp);   //将边数按从小到大排序
for(i=0;i<n;i++)   //采用Kruskal算法  每个点指向自己
root[i]=i;

/*for(i=0;i<k;i++)      //输出查看
printf("%d--%d  %d ",s[i].x,s[i].y,s[i].price);
printf("\n");

for(i=0;i<n;i++)
printf("%d->%d ",i,root[i]);
printf("\n--------------------------------\n");*/

int w;
for(i=0;i<k&&js<n;i++)
if(root[s[i].x]!=root[s[i].y]){
total+=s[i].price;
root[root[s[i].y]]=root[s[i].x];   //合并两棵树
for(j=0;j<n;j++){     //更新祖先节点
w=j;
while(root[w]!=w)
w=root[w];
root[j]=w;
}
js++;
}

return total;
}
int main(){
int n,i,t=1,z[101];

//freopen("Jungle Roads.out","w",stdout);   //文件输出

scanf("%d",&n);
while(n!=0){
z[t++]=deal(n);   //z[t]表示每次处理后的费用
scanf("%d",&n);

}

for(i=1;i<t;i++)
printf("%d\n",z[i]);

return 0;
}