题意比较好懂,就不说啦~
思路:建图,求最小生成树,两个坑点就是注意重边和注意图不连通的情况。
代码如下:
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <map>
#include <vector>
#include <set>
#include <string>
#define PB push_back
#define FT first
#define SD second
#define MP make_pair
#define INF (0x3f3f3f3f)*2
using namespace std;
typedef long long LL;
typedef pair<int,int> P;
const int maxn=105,MOD=7+1e9;
map<string,int> mp;
char s1[maxn],s2[maxn];
int ans,N,tot;
int dis[maxn],fa[maxn];
bool vis[maxn];
struct node
{
int v,w,next;
}e[maxn];
void init()
{
N=tot=ans=0;
mp.clear();
memset(fa,-1,sizeof(fa));
memset(vis,0,sizeof(vis));
}
int readin(string ss)
{
if(mp.find(ss)==mp.end()) {
mp[ss]=N++;
}
return mp[ss];
}
void addedge(int u,int v,int w){
e[tot].v=v;
e[tot].w=w;
e[tot].next=fa[u];
fa[u]=tot++;
}
bool MST()
{
dis[0]=0;
for(int i=1;i<N;i++)
dis[i]=INF;
for(int i=fa[0];i!=-1;i=e[i].next){
dis[e[i].v]=min(e[i].w,dis[e[i].v]);
}
vis[0]=1;
for(int _=0;_<N;_++){
int _min=INF,u=-1;
for(int i=0;i<N;i++){
if(!vis[i]&&_min>dis[i]){
_min=dis[i];
u=i;
}
}
if(u==-1) break;
vis[u]=1;
ans+=_min;
for(int i=fa[u];i!=-1;i=e[i].next){
int v=e[i].v,w=e[i].w;
if(!vis[v]&&dis[v]>w) dis[v]=w;
}
}
for(int i=0;i<N;i++)
if(dis[i]==INF) return 0;
return 1;
}
int main()
{
//freopen("E:/ACM_code/in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int T,ca=0;
scanf("%d",&T);
while(T--){
init();
int m,w;
scanf("%d",&m);
for(int i=1;i<=m;i++){
scanf("%s%s%d",s1,s2,&w);
int u=readin((string)s1),v=readin((string)s2);
addedge(u,v,w),addedge(v,u,w);
}
printf("Case %d: ",++ca);
MST()?printf("%d\n",ans):puts("Impossible");
}
//system("pause");
return 0;
}