hdu2112 Today

主要用map将字符串转化为标号就可以直接套个模板了

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<map>
#define inf  999999999
#define maxn 155
using namespace std;

int g[maxn][maxn];
int dist[maxn];
int vis[maxn];
int n,m;

void dijkstra(int st){
	int i,j,mindis,u;
	memset(vis,0,sizeof(vis));
	for(i=1;i<=n;i++){
		dist[i]=g[st][i];
	}
	vis[st]=1;
	for(i=1;i<=n;i++){
		mindis=inf;
		u=-1;
		for(j=1;j<=n;j++){
			if(!vis[j]&&mindis>dist[j]){
				mindis=dist[j];
				u=j;
			}
		}
		if(u==-1)break;
		vis[u]=1;
		for(j=1;j<=n;j++){
			if(!vis[j]&&dist[j]>dist[u]+g[u][j]){
				dist[j]=dist[u]+g[u][j];
			}
		}
	}
}
int main(){
	int i,j;
	while(scanf("%d",&m)!=EOF){
		if(m==-1)break;
		for(i=1;i<maxn;i++){
			for(j=1;j<maxn;j++){
				if(i==j)g[i][j]=0;
				else g[i][j]=inf;
			}
			dist[i]=inf;
		}
		map<string,int>mp;
		char st[50],ed[50];
		scanf("%s%s",st,ed);
		mp[st]=1;
		mp[ed]=2;
		int num=3;
		for(i=0;i<m;i++){
			char a[50],b[50];
			int x;
			scanf("%s%s%d",a,b,&x);
			if(!mp[a])mp[a]=num++;
			if(!mp[b])mp[b]=num++;
			if(g[mp[a]][mp[b]]>x){
				g[mp[a]][mp[b]]=g[mp[b]][mp[a]]=x;	
			}
		}
		if(strcmp(st,ed)==0){//起点终点相同的时候特判一下
			printf("0\n");
			continue;
		}
		n=num-1;
		dijkstra(1);
		if(dist[2]==inf)printf("-1\n");//无法到达的情况记得判断
		else printf("%d\n",dist[2]);
	}
	return 0;
}