HDU 2112 HDU Today

思路:最短路问题map+floyd

本来想到用字典树来处理站点的问题，不过想来map要容易得多

#include<iostream>
#include<map>
#include<string>
using namespace std;
const int INF=1<<28;
const int N=155;
int road[N][N];
map<string,int> v;
void Init()
{
   int i,j;
   for(i=1;i<155;i++)
	   for(j=1;j<155;j++)
	   {
	      if(i==j) road[i][j]=0;
		  else road[i][j]=road[j][i]=INF;
	   }

}
int floyd(int n)
{
  int i,j,k;
  for(k=1;k<n;k++)
	  for(i=1;i<n;i++)
	  {
	     if(road[i][k]!=INF)
		 {
		    for(j=1;j<n;j++)
			   if(road[i][j]>road[i][k]+road[k][j])
			      road[i][j]=road[i][k]+road[k][j];
		 }
	  }
   return road[1][2];
}
int main()
{
   int n,i,j,flag,kind,p;
   char start[55],end[55],s[55],e[55];
   while(scanf("%d",&n)!=EOF&&(n!=-1))
   {
	   v.clear();flag=0;
	   Init();kind=3;
       scanf("%s",start);
	   scanf("%s",end);
	   v[start]=1;v[end]=2;
	   if(strcmp(start,end)==0) flag=1;//始终点一样要特殊处理
	   for(i=0;i<n;i++)
	   {
	     scanf("%s",s);
	     scanf("%s",e);
		 if(!v[s]) v[s]=kind++;
		 if(!v[e]) v[e]=kind++;
		 scanf("%d",&p);
		 if(p<road[v[s] ][v[e] ])
			 road[v[s] ][v[e] ]=road[v[e] ][v[s] ]=p;
	   }
	   if(flag)
	   {
		   printf("0\n");
		   continue;
	   }
	   int ans=floyd(kind);
	   printf("%d\n",ans==INF? -1:ans);
	   
   
   }
   return 0;
}