代码难度普及+,思维难度提高+。
但事实上如果想不到硬干很难搞出来。
我们要求这些点对中最近的点。
如果直接以每个点为起点跑单源最短路是肯定爆炸的。
所以我们换一个角度。我们不用特殊点来更新答案。我们用边作为桥梁,用普通点作为桥墩来更新答案。
人话说,如果有一条边连接a,b两个点,x是可以到a的最近特殊点,y是b能到的最近特殊点,那x和y的最短距离就可能是dis(x,a)+w(a,b)+dis(b,y)。
这样我们就通过中间的普通点边完成了更新答案。
同时我们发现,特殊点是谁并不重要,我们只需要距离。
所以跑两遍最短路。第一遍求出能到点的最近特殊点,第二遍求出点能到的最近特殊点。
可以发现第二遍只需要把图反过来就能照样跑。
使用vector存边可以很好反转。如此我们得到了一个优秀的算法。复杂度还是nlogn。
#include<bits/stdc++.h>
using namespace std;
#define in read()
#define int long long
int in{
int cnt=0,f=1;char ch=0;
while(!isdigit(ch)){
ch=getchar();if(ch=='-')f=-1;
}
while(isdigit(ch)){
cnt=cnt*10+ch-48;
ch=getchar();
}return cnt*f;
}
vector<pair<int,int> > g[100003],rg[100003];
int dis1[100003],dis2[100003],num1[100003],num2[100003];
int t,n,m,k;
int tag[100003];
int sp[100003];
priority_queue<pair<int,int> >q;
int vis[100003];
void dij(int *dis,int *num){
for(int i=1;i<=n;i++)dis[i]=1e9+7;
memset(vis,0,sizeof(vis));
for(int i=1;i<=k;i++){
dis[sp[i]]=0;num[sp[i]]=sp[i];q.push(make_pair(0,sp[i]));
}
while(!q.empty()){
int u=q.top().second;q.pop();if(vis[u])continue;vis[u]=1;
for(int i=0;i<g[u].size();i++){
int v=g[u][i].first;//cout<<" ## "<<u<<" "<<v<<endl;
//cout<<dis[v]<<" "<<dis[u]+g[u][i].second<<endl;
if(dis[v]>dis[u]+g[u][i].second){
dis[v]=dis[u]+g[u][i].second;
num[v]=num[u];q.push(make_pair(-dis[v],v));
}
}
}
}
int ans;
signed main(){
t=in;
while(t--){
ans=1e18;
n=in;m=in;k=in;
for(int i=1;i<=m;i++){
int a,b,c;a=in;b=in;c=in;
g[a].push_back(make_pair(b,c));
rg[b].push_back(make_pair(a,c));
}
for(int i=1;i<=k;i++){
sp[i]=in;tag[sp[i]]=1;
}
dij(dis1,num1);
for(int i=1;i<=n;i++)g[i]=rg[i];
dij(dis2,num2);
// for(int i=1;i<=n;i++){
// cout<<dis1[i]<<" "<<num1[i]<<" "<<dis2[i]<<" "<<num2[i]<<endl;
// }
for(int i=1;i<=n;i++){
for(int j=0;j<g[i].size();j++){
if(num1[g[i][j].first]!=num2[i]&&num1[g[i][j].first]&&num2[i]){
//cout<<"#";
ans=min(ans,dis1[g[i][j].first]+dis2[i]+g[i][j].second);
}
}
}
printf("%lld\n",ans);
for(int i=1;i<=n;i++){
g[i].clear();rg[i].clear();tag[i]=0;num1[i]=num2[i]=0;
}
}
return 0;
}