Light OJ 1041 Road Construction

  题意比较好懂，就不说啦~

  思路：建图，求最小生成树，两个坑点就是注意重边和注意图不连通的情况。

代码如下：

#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;
}