1006

来源:互联网 发布:婚礼入场音乐知乎 编辑:程序博客网 时间:2024/05/01 08:07

题意:
给出城市及几个城市之间的距离,求最小生成树。

思路:
用prim算法比较好

感想:

水题。

代码:

#include<iostream>

#include<cstdio>
#include<algorithm>
using namespace std;
char p[1000];
int n,k;
struct road
{
char s,e;
int w;
}r[1000];
bool cmp(road a,road b)

{ return a.w<b.w; }

char find(char x)
{ return p[x]==x?x:p[x]=find(p[x]); }
int kru()
{
int sum=0;
for(int i=0;i<k;i++)
{
char x=find(r[i].s),y=find(r[i].e);
if(x!=y)
{
p[x]=y;
sum+=r[i].w;
}
}
return sum;
}
int main()
{
while(cin>>n&&n)
{
char rs,re;
int m,rw;
k=0;
n--;
while(n--)
{
cin>>rs>>m;
while(m--)
{
cin>>re>>rw;
r[k].s=rs;
r[k].e=re;
r[k].w=rw;
k++;
}
}
for(char i='A';i<='Z';i++)
p[i]=i;
sort(r,r+k,cmp);
cout<<kru()<<endl;
}
return 0;
}
0 0
原创粉丝点击