专题四——1006

来源:互联网 发布:东方快车杀人事件知乎 编辑:程序博客网 时间:2024/04/30 13:32

题目:维护公路网


题意:
  热带岛屿中的村庄,因为被丛林阻挡道路,所以需要每月去维护。并且给出限制条件,公路网中的路最大不能超过75条,没有一个村庄会超过15条路与其他村庄相连通。现在需要编写一个程序,求出每月维护公路网的最小成本。村庄用英文字母表示

思路:
  求出最小生成树的问题。
  Prim算法或Kruskal算法。

感想:
需要并查集的思想

代码:
#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