如果按照正常的思路,对每个起点进行dijkstra算法的话,会超时。由于终点只有一个,这时可以考虑将终点看做起点,起点看做多个终点,这样就可以一次dijkstra,得出答案。
详见代码:
/*************************************************************************
> File Name: main.cpp
> Author:Eagles
> Mail:None
> Created Time: 2018年09月07日 星期五 18时11分35秒
> Description:HDU2680
************************************************************************/
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
#define N 10005
struct node
{
int to;
int val;
int nex;
}E[200005];
int head[N];
int dis[N];
bool vis[N];
int n,m,cnt;
int num;
int star;
int en[N];
void addEdge(int a, int b, int val)//注意,是单向边!!
{
E[cnt].to=b;
E[cnt].val=val;
E[cnt].nex=head[a];
head[a]=cnt++;
}
void init()
{
memset(head,-1,sizeof(head));
memset(vis,false,sizeof(vis));
memset(dis,0x3f3f3f3f,sizeof(dis));
cnt=0;
star--;
for (int i=0; i<m; i++)
{
int a,b,val;
scanf("%d%d%d",&a,&b,&val);
addEdge(b-1,a-1,val);//换一下顺序
}
scanf("%d",&num);
for (int i=0; i<num; i++)
{
int a;
scanf("%d",&a);
en[i]=a-1;
}
int k=head[star];
while (k != -1)
{
if (dis[E[k].to]>E[k].val)
dis[E[k].to]=E[k].val;
k=E[k].nex;
}
dis[star]=0;
vis[star]=true;
}
void dijkstra()
{
for (int i=0; i<n-1; i++)
{
int minn=123456;
int tmp;
for (int j=0; j<n; j++)
{
if (!vis[j]&&dis[j]<minn)
{
minn=dis[j];
tmp=j;
}
}
vis[tmp]=true;
int k=head[tmp];
while (k != -1)
{
if (!vis[E[k].to]&&dis[E[k].to]>dis[tmp]+E[k].val)
dis[E[k].to]=dis[tmp]+E[k].val;
k=E[k].nex;
}
}
}
void get_ans()
{
int ans=0x3f3f3f3f+5;
for (int i=0; i<num; i++)
ans=ans<dis[en[i]]?ans:dis[en[i]];
if (ans == 0x3f3f3f3f)
ans=-1;
printf("%d\n",ans);
}
int main()
{
while (~scanf("%d%d%d",&n,&m,&star))
{
init();
dijkstra();
get_ans();
}
return 0;
}