莫欺少年穷,修魔之旅在这开始—>算法提高课题解
思路:
1. 核心:建立虚拟原点
2. 从虚拟原点出发用 spfa 求最短路
#include<bits/stdc++.h>
using namespace std;
const int N = 1010, M = 21010, INF = 0x3f3f3f3f;
int n,m,s,t;
int h[N],e[M],w[M],ne[M],idx;
int dist[N];
bool st[N];
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
//简单的 spfa 求最短路
int spfa()
{
//多组数据,st 要初始化
memset(st,0,sizeof st);
memset(dist,0x3f,sizeof dist);
dist[0]=0;
queue<int> q;
q.push(0);
while(q.size())
{
auto 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])
{
q.push(j);
st[j]=true;
}
}
}
}
if(dist[s]==INF) return -1;
return dist[s];
}
int main()
{
while(cin>>n>>m>>s)
{
memset(h,-1,sizeof h);
idx=0;
while(m--)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
}
cin>>t;
//建立虚拟原点0
while(t--)
{
int ver;
cin>>ver;
add(0,ver,0);
}
cout<<spfa()<<endl;
}
return 0;
}