一个人的旅行 (源点建图)
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
输出草儿能去某个喜欢的城市的最短时间。
我们不可能枚举起点来找最短距离,肯定会T,所以我们加入一个源点,编号1002,到所有 Si 的距离为0,那么当以源点为起点跑Djikstra后,dis[i]中就是到每个Si 的最短距离。 注意到虽然题目给的城市编号可能不连续,所以我们把n=1002即可;
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int INF = 0x3f3f3f3f;
const int Y = 1002;
const int N = 1e3+7;
int n;
vector<PII> G[N];
int dis[N];
bool vis[N];
void Dijkstra(int s)
{
memset(dis,0x3f,sizeof dis);
memset(vis,0,sizeof vis);
dis[s] = 0;
for(int i=1;i<=n;i++){
int t = -1;
for(int j=1;j<=n;j++)
if(!vis[j]&&(t==-1||dis[t]>dis[j]))
t = j;
vis[t] = 1;
for(int j=0;j<G[t].size();j++){
int x = G[t][j].first, w = G[t][j].second;
dis[x] = min(dis[x],dis[t]+w);
}
}
}
int main()
{
int T,s,d;
while(~scanf("%d%d%d",&T,&s,&d)){
int a,b,c;
while(T--){
scanf("%d%d%d",&a,&b,&c); //邻接矩阵存图时注意处理重边;
G[a].push_back({b,c});
G[b].push_back({a,c});
}
while(s--){
scanf("%d",&a);
G[Y].push_back({a,0});
G[a].push_back({Y,0});
}
Dijkstra(Y);
int ans=INF;
while(d--){
scanf("%d",&a);
ans = min(ans,dis[a]);
}
printf("%d\n",ans);
for(int i=0;i<=n;i++) G[i].clear();
}
return 0;
}