Problem Description
虽然草儿是个路痴(就是在杭电待了一年多,居然还会在校园里迷路的人,汗~),但是草儿仍然很喜欢旅行,因为在旅途中 会遇见很多人(白马王子,^0^),很多事,还能丰富自己的阅历,还可以看美丽的风景……草儿想去很多地方,她想要去东京铁塔看夜景,去威尼斯看电影,去阳明山上看海芋,去纽约纯粹看雪景,去巴黎喝咖啡写信,去北京探望孟姜女……眼看寒假就快到了,这么一大段时间,可不能浪费啊,一定要给自己好好的放个假,可是也不能荒废了训练啊,所以草儿决定在要在最短的时间去一个自己想去的地方!因为草儿的家在一个小镇上,没有火车经过,所以她只能去邻近的城市坐火车(好可怜啊~)。
Input
输入数据有多组,每组的第一行是三个整数T,S和D,表示有T条路,和草儿家相邻的城市的有S个,草儿想去的地方有D个;
接着有T行,每行有三个整数a,b,time,表示a,b城市之间的车程是time小时;(1=<(a,b)<=1000;a,b 之间可能有多条路)
接着的第T+1行有S个数,表示和草儿家相连的城市;
接着的第T+2行有D个数,表示草儿想去地方
Output
输出草儿能去某个喜欢的城市的最短时间。
Sample Inpu
6 2 3 1 3 5 1 4 7 2 8 12 3 8 4 4 9 12 9 10 2 1 2 8 9 10
Sample Output
9
思路
因为有多个原点和起点,一开始想到的是用floyd算法,但是点数最多有1000个,三重循环应该会超时,所以用超级原点和超级终点的dijsktra算法的画只需要一遍就可以解决。
没有使用堆优化的dijsktra也可以过。
杭电的oj好多题都是需要多组数据输入的...因为没看题wa了好多次才发现。
#include<bits/stdc++.h>
using namespace std;
int mp[1005][1005];
int dis[1005];
int vis[1005];
int main(){
int t,s,d;
while(cin>>t>>s>>d){
memset(mp,0x1f,sizeof(mp));
memset(vis,0,sizeof(vis));
memset(dis,0x1f,sizeof(dis));
int maxn=-1;
for(int i=0;i<t;i++){
int a,b,time;
cin>>a>>b>>time;
mp[a][b]=mp[b][a]=min(time,mp[a][b]);
maxn=max(maxn,max(a,b));
}
for(int i=0;i<s;i++){
int a;
cin>>a;
mp[0][a]=0;
}
for(int i=0;i<d;i++){
int b;
cin>>b;
mp[b][maxn+1]=0;
}
int k=maxn+1;
for(int i=1;i<=k;i++){
dis[i]=mp[0][i];
}
for(int i=1;i<=k;i++){
int index=-1,minn=1e9;
for(int j=1;j<=k;j++){
if(minn>dis[j]&&!vis[j]){
minn=dis[j];
index=j;
}
}
vis[index]=1;
for(int j=1;j<=k;j++){
if(!vis[j]&&dis[j]>dis[index]+mp[index][j]){
dis[j]=dis[index]+mp[index][j];
}
}
}
cout << dis[k] <<endl;
}
return 0;
}