又是一道最短路,如果普通的按照从w个点到s的最短距离中选择最短的会超时,因为w可能很接近n.可以把所有的边反一下就变成了求从s点到w个点最短距离中的最短距离,复杂度一下子就下来了.
参考了网上还有一种方法:设一个超级原点,把这个超级原点和w个起点连边,权值为0,然后求超级原点到s的最短路就是答案的,很巧妙的方法
#include <iostream>
#include <cstdio>
#include <memory.h>
#include <queue>
#include <vector>
using namespace std;
const int maxn=1010;
struct Edge
{
int v,w,next;
}e[20010];
struct node
{
int i,v;
node(int ii=0,int vv=0):i(ii),v(vv){}
bool operator<(const node &rhs)const{
return v>rhs.v;
}
};
int head[maxn],dis[maxn],n,m,s,q;
bool vis[maxn];
vector<int>v;
int dijkstra(int s){
memset(dis,0X7F,sizeof(dis));
memset(vis,0,sizeof(vis));
dis[s]=0;
priority_queue<node>q;
q.push(node(s,0));
while (q.size()){
node tn=q.top();
q.pop();
if(vis[tn.i])continue;
for (int ne=head[tn.i];ne!=-1;ne=e[ne].next)
{
int v=e[ne].v,w=e[ne].w;
if (dis[v]>dis[tn.i]+w&&!vis[v])
{
dis[v]=dis[tn.i]+w;
q.push(node(v,dis[v]));
}
}
}
int ans=0X7F7F7F7F;
for (int i=0;i<v.size();++i)
{
ans=min(ans,dis[v[i]]);
}
return ans==0X7F7F7F7F?-1:ans;
}
int main(){
while (scanf("%d%d%d",&n,&m,&s)==3)
{
v.clear();
for (int i=1;i<=n;++i){
head[i]=-1;
}
for (int i=0;i<m;++i){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
e[i].v=u;//把边方向取反
e[i].w=w;
e[i].next=head[v];
head[v]=i;
}
scanf("%d",&q);
while (q--)
{
int s;
scanf("%d",&s);
v.push_back(s);
}
printf("%d\n",dijkstra(s));
}
return 0;
}