hdu 2112 HDU Today Dijkstra算法应用+set集合判重+二分查找。不错的一题

HDU Today

Time Limit: 15000/5000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 16491    Accepted Submission(s): 3905

Problem Description

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

 

Input

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

 

Output

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

 

Sample Input

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

 

Sample Output

50


Hint:
The best route is:
xiasha->ShoppingCenterofHangZhou->supermarket->westlake


虽然偶尔会迷路，但是因为有了你的帮助
**和**从此还是过上了幸福的生活。

――全剧终――

 

Author

lgx

 

这道题用Dijkstra处理的部分大家应该都会，主要就是字符串对应的下标不好求，，其实这道题就这点比较有趣，方法很多，有的人用map，还可以用Hash,我自己用的是set集合。这道题其实用map应该是最简单的。

代码很容易理解，看代码：

#include <iostream>
#include <string>
#include <algorithm>
#include <set>
#include <cstring>
#define MAX 10100
#define INF 500000000
using namespace std ;

bool cmp(const string &a ,const string &b)
{
	return a>b;
}
string place[MAX] ; 
int dis[MAX] , graph[MAX][MAX] ;
bool closed[MAX] ;
//二分查找 
int Bsearch(int n,const string &a)
{
	int s = 1 , e = n;
	while(s<=e)
	{
		int mid = (s+e)>>1;
		if(place[mid]>a)
		{
			s = mid+1;
		}
		else if(place[mid]<a)
		{
			e = mid-1;
		}
		else
		{
			return mid ;
		}
	}
	return -1;
}
// dijkstra算法 
void dijkstra(int u , int k)
{
	memset(closed,false,sizeof(closed)) ;
	for(int i = 1 ; i<=k ; ++i)
	{
		dis[i] = graph[u][i] ;
	}
	closed[u] = true ; 
	dis[u] = 0 ;	
	for(int i = 0 ; i < k ; ++i)
	{
		int min = INF , index = 0 ;
		for(int j = 1 ; j <= k ; ++j)
		{
			if(!closed[j] && min>dis[j])
			{
				min = dis[j];
				index = j ;
			}
		}
		if(min == INF)
		{
			break ;
		}
		closed[index] = true ;
		for(int j = 1 ; j <= k ; ++j)
		{
			if(dis[j]>dis[index]+graph[index][j])
			{
				dis[j]=dis[index]+graph[index][j] ;
			}
		}
	}
}

int main()
{
	string strA[MAX],strB[MAX];
	int n ,w[MAX];
	set<string> s ;
	while(cin>>n && n!=-1)
	{
		string start, end ;
		cin>>start>>end ;
		for(int i = 0 ; i < n ; ++i)
		{
			cin>>strA[i]>>strB[i]>>w[i] ;
			s.insert(strA[i]);
			s.insert(strB[i]);
		}
		set<string>::iterator it ;
		int k = 1 ;
		for(it = s.begin() ; it != s.end() ; ++it)
		{
			place[k++] = *it ;
		}
		s.clear() ;
		sort(place+1,place+k,cmp) ;	//排序是为了O(logn)的查找，是为了对应下标 
		for(int i = 0 ; i < k+10 ; ++i)
		{
			for(int j = 0 ; j <= i ; ++j)
			{
				graph[i][j] = INF ;
				graph[j][i] = INF ;
			}
		}
		for(int i = 0 ; i < n ; ++i)
		{
			int x = Bsearch(k,strA[i]) ;
			int y = Bsearch(k,strB[i]) ;
			if(graph[x][y]>w[i])		//判重边，这道题不知道有没有重边，但是做图论的题，最好都判断一下 
			{
				graph[x][y] = graph[y][x] = w[i] ;
			}
		}
		int x = Bsearch(k,start);
		int y = Bsearch(k,end) ;
		if(x == -1 || y == -1)		//说明起始地与目的地所在的子图不连通。 
		{
			cout<<"-1"<<endl;
			continue ; 
		}
		dijkstra(x,k) ;
		if(dis[y] == INF)
		{
			cout<<"-1"<<endl;
		}
		else
		{
			cout<<dis[y]<<endl ;
		}
	}
	return 0 ;
}
/*
附送简单的测试数据 
6
xiasha west
xiasha station 60
xiasha ShoppingCenterofHangZhou 30
station westlake 20
ShoppingCenterofHangZhou supermarket 10
xiasha supermarket 50
supermarket westlake 10
-1

9
xiasha west
xiasha station 60
xiasha ShoppingCenterofHangZhou 30
station westlake 20
ShoppingCenterofHangZhou supermarket 10
xiasha supermarket 50
supermarket westlake 10
fujian west 20
wet lanzhou 16
wet nanjin 28
-1

*/