Hdoj 2112

原题链接

描述

经过锦囊相助，海东集团终于度过了危机，从此，HDU的发展就一直顺风顺水，到了2050年，集团已经相当规模了，据说进入了钱江肉丝经济开发区500强。这时候，XHD夫妇也退居了二线，并在风景秀美的诸暨市浬浦镇陶姚村买了个房子，开始安度晚年了。
这样住了一段时间，徐总对当地的交通还是不太了解。有时很郁闷，想去一个地方又不知道应该乘什么公交车，在什么地方转车，在什么地方下车（其实徐总自己有车，却一定要与民同乐，这就是徐总的性格）。
徐总经常会问蹩脚的英文问路：“Can you help me?”。看着他那迷茫而又无助的眼神，热心的你能帮帮他吗？
请帮助他用最短的时间到达目的地（假设每一路公交车都只在起点站和终点站停，而且随时都会开）。

输入

输入数据有多组，每组的第一行是公交车的总数N(0<=N<=10000)；
第二行有徐总的所在地start，他的目的地end；
接着有n行，每行有站名s，站名e，以及从s到e的时间整数t(0<t<100)(每个地名是一个长度不超过30的字符串)。
note：一组数据中地名数不会超过150个。
如果N==-1，表示输入结束。

输出

如果徐总能到达目的地，输出最短的时间；否则，输出“-1”。

样例输入

6
xiasha westlake
xiasha station 60
xiasha ShoppingCenterofHangZhou 30
station westlake 20
ShoppingCenterofHangZhou supermarket 10
xiasha supermarket 50
supermarket westlake 10
-1

样例输出

50

思路

dijkstra算法，需要稍微优化一下，不然会超时。
读入的时候注意一下，我选择用map，然后编号从1开始。
我刚开始理解错题意，以为公交车站有N个，然后WA了好几次。

代码

#include <bits/stdc++.h>
#define INF 99999999
#define maxn 152
using namespace std;  

int dist[maxn][maxn];
int length[maxn], f[maxn];
int n, sum;
char s1[50], s2[50];

void init()
{
	sum = 2;
	memset(f, 0, sizeof(f));
	for(int i = 0; i < maxn; i++)
		for(int j = 0; j < maxn; j++)
			dist[i][j] = INF;
}

void dijkstra(int v)
{
	f[v] = 1;
	for(int i = 1; i <= sum; i++)
		length[i] = dist[v][i];
	while(1)
	{
		int min = INF, k = -1;
        for(int j = 1; j <= sum; j++)
            if(!f[j] && length[j] < min) {min = length[j]; k = j;}
        if(k == -1) break;
        f[k] = 1;
        for(int j = 1; j <= sum; j++)
            if(!f[j] && dist[k][j] < length[j] - min) length[j] = dist[k][j] + min;
	}
}

int main()
{
    while(~scanf("%d", &n))
    {
    	if(n == -1) break;
    	init();
    	scanf("%s %s", s1, s2);
    	map<string, int> mp;
    	mp[s1] = 1; mp[s2] = 2;
    	if(strcmp(s1,s2) == 0) dist[1][2] = 0;
    	for(int i = 0; i < n; i++)
    	{
    		int d;
    		scanf("%s %s %d", s1, s2, &d);
    		if(mp[s1] == 0) mp[s1] = ++sum;
    		if(mp[s2] == 0) mp[s2] = ++sum;
    		if(dist[mp[s1]][mp[s2]] > d)
    			dist[mp[s1]][mp[s2]] = dist[mp[s2]][mp[s1]] = d;
		}
		dijkstra(1);
		if(length[2] < INF) printf("%d\n", length[2]);
		else printf("-1\n");
	}
    return 0;  
}