最短路入门HDU2112

http://acm.hdu.edu.cn/showproblem.php?pid=2112

中文题。

map可以水的吧。  字典树+spfa+邻接表。  数组开的有点大 ，复杂度其实可以降不少。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <string>
#include <iostream>
using namespace std;
const int INF=0xfffffff;
struct edge
{
	int to;int val;int next;
}e[1111111];

int l=0;
int head[22222];
void add(int from ,int to,int val)
{
	e[l].to=to;
	e[l].val=val;
	e[l].next=head[from];
	head[from]=l++;
}
int dis[1111111];
int vis[1111111];
void spfa(int x)
{
	for(int i=0;i<1000000;i++)
		dis[i]=INF;
	memset(vis,0,sizeof(vis));
	dis[x]=0;vis[x]=1;
	queue<int> q;
	q.push(x);
	while(!q.empty()){
		int cur=q.front();
		q.pop();
		vis[cur]=0;
		for(int i=head[cur];i!=-1;i=e[i].next){
			int cc=e[i].to;
			if(dis[cc]>dis[cur]+e[i].val){
				dis[cc]=dis[cur]+e[i].val;
				if(!vis[cc]){
					vis[cc]=1;
					q.push(cc);
				}
			}
		}
	}
}

struct Node
{
	int next[59];
}node[1111111];

int ret=0;

int Insert(char *s)
{
	int len=strlen(s);
	int root=0;
	for(int i=0;i<len;i++){
		int cc=s[i]-'a';
		if(!node[root].next[cc]){
			node[root].next[cc]=++ret;
		}
		root=node[root].next[cc];
	}
	return root;
}

int Find(char *s)
{
	int len=strlen(s);
	int root=0;
	for(int i=0;i<len;i++){
		int cc=s[i]-'a';
		root=node[root].next[cc];
	}
	return root;
}

int main()
{
	int n;char str3[2000],str4[2000],str[2000],str1[2000];
	int a,b,val;
	while(scanf("%d",&n),n!=-1){
		memset(head,-1,sizeof(head));
		scanf("%s%s",str3,str4);
		for(int i=0;i<n;i++){
			scanf("%s%s",str,str1);
			scanf("%d",&val);
			int a=Insert(str);int b=Insert(str1);
			add(a,b,val);
			add(b,a,val);
		}
		a=Find(str3);b=Find(str4);
		spfa(a);
		if(dis[b]==INF)
		printf("-1\n");
		else
			printf("%d\n",dis[b]);
	}
	return 0;
}