Tangled in Cables（Kruskal+map容器处理字符串）

/**

题意：

给你两个城市之间的道路（无向图），求出需要的

电缆。如果大于所提供的，就输出Not enough 。。。

否则输出所需要的电缆长度。

输入：N （给定的电缆总长度）

m1 （有多少个城市—）

str1

str2

str3

str4

:

；（城市的名字）

m2（相当于给出m2条边）

a b c （a城市到b城市的距离为c）

：

：

：

:

输出：

所需的最短长度 若大于给定的长度 输出 Not enough cable

分析： 简单的最小生成树+字符串处理

用map容器映射处理简单

*/

#include<stdio.h>

#include<map>

#include<algorithm>

#include<string>

#include<string.h>

#include<iostream>

using namespace std;

const int MAX=1e4;///1*10 的4次方

int vest[MAX];

int n,m;

double leng=0;

map<string,int >mp;

struct node

{

int u,v;

double w;

} bian[MAX];

bool cmp(node a,node b)

{

return a.w<b.w;

}

void init(int n)

{

for(int i=0; i<=n; i++)

vest[i]=i;

}

int Find(int t)

{

if(vest[t]==t)return t;

return Find(vest[t]);

}

bool merge(int a,int b)

{

int x=Find(a);

int y=Find(b);

if(x!=y)

{

vest[x]=y;

return true;

}

return false;

}

double Kruskal()

{

double sum=0;

int cnt=0;

for(int i=0; i<m; i++)

{

if(merge(bian[i].u,bian[i].v))

{

sum+=bian[i].w;

++cnt;

}

if(cnt>=n-1)break;

}

return sum;

}

int main()

{

scanf("%lf",&leng);

{

mp.clear();

scanf("%d",&n);

init(n);

char a[25];

int k=1;

for(int i=0; i<n; i++)

{

scanf("%s",a);

if(mp[a]==0)mp[a]=k++;///给第一次出现的城市编号

}

scanf("%d",&m);

char b[25];

double c;

for(int i=0; i<m; i++)

{

scanf("%s%s%lf",a,b,&c);

bian[i].u=mp[a];

bian[i].v=mp[b];

bian[i].w=c;

}

sort(bian,bian+m,cmp);

double sum=Kruskal();

if(sum>=leng)printf("Not enough cable\n");

else printf("Need %.1lf miles of cable\n",sum);

}

return 0;

}