单源最短路扩展:创建虚拟源点
解题思路:题目大致意思是求在题目给出的点中离s最近的距离,其实我们可以创建一个虚拟源点让各个点到他的路径是0,spfa 0 就行了。
#include<bits/stdc++.h>
using namespace std;
const int N=1010,M = 50010;
int e[M],h[N],idx,ne[M],w[M];
int dist[N];
bool st[N];
int n , m , s ;
void add(int a,int b,int c)
{
w[idx] = c ,ne[idx]=h[a],e[idx]=b ,h[a]= idx++;
}
int spfa()
{
memset(st,0,sizeof st);
memset(dist,0x3f,sizeof dist);
dist[0]=0;
queue<int>q;
q.push(0);
st[0]=true;
while(q.size())
{
int t=q.front();
q.pop();
st[t]=false;
for(int i=h[t];~i;i=ne[i])
{
int j= e[i];
if(dist[j]>dist[t]+w[i])
{
dist[j]=dist[t]+w[i];
if(!st[j])
{
st[j]=true;
q.push(j);
}
}
}
}
if(dist[s]==0x3f3f3f3f) return -1;
return dist[s];
}
int main()
{
while(cin >> n >> m >>s)
{
idx = 0;
memset(h,-1,sizeof h);
while(m--)
{
int a,b,c;
cin >> a >> b >>c;
add(a,b,c);
}
int t;
cin >> t;
while(t--)
{
int x;
cin >>x;
add(0,x,0);
}
cout << spfa()<<endl;
}
}