这题就是最小生成树的模板题,可是这题的数据给的有点坑。
首先我不清楚给出的村庄是否就是按照从A开始的字母顺序排列,所以我就用一个exist布尔数组来标记这个村庄是否存在;
在者就是两个村庄之间可能有多条路,所以存储的时候应该存储最小的,可是题目没有这样的例子,哎!
#include<cstdio> #include<iostream> #include<cstring> #include<algorithm> #include<stack> #include<queue> using namespace std; const int inf=1<<28; int road[27][27],mincost[27]; bool used[27],exist[27]; int n; int solve() { fill(mincost,mincost+27,inf); memset(used,false,sizeof(used)); for(int i=1;i<=26;i++) if(exist[i]) { mincost[1]=0; break; } int cost=0; while(true) { int _min=inf,u=-1; for(int i=1;i<=26;i++) if(exist[i]&&!used[i]&&_min>mincost[i]) _min=mincost[i],u=i; if(u==-1) break; used[u]=true; cost+=_min; for(int i=1;i<=n;i++) if(exist[i]) mincost[i]=min(mincost[i],road[u][i]); } return cost; } int main() { while(scanf("%d",&n),n!=0) { char s1[2],s2[2]; for(int i=0;i<27;i++) for(int j=0;j<27;j++) road[i][j]=inf; int m,k,a,b; memset(exist,false,sizeof(exist)); for(int i=1;i<n;i++) { scanf("%s%d",s1,&m); a=(s1[0]-'A')+1; exist[a]=true; for(int j=1;j<=m;j++) { scanf("%s%d",s2,&k); b=(s2[0]-'A')+1; if(exist[b]&&k<road[a][b]) road[a][b]=road[b][a]=k; else { exist[b]=true; road[a][b]=road[b][a]=k; } } } int ans=solve(); printf("%d\n",ans); } return 0; }
时间: 2024-10-17 14:22:02