不是所有奶牛行走的总距离,将每个牧场看作只有一个牛 做n次dijkstra记得每次初始化st和dist
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef pair<int,int> PII;
const int N=1000,M=3000,INF=0x3f3f3f3f;
int dist[N],h[N],e[M],ne[M],w[M],idx,cow[N];
int n,m,p;
bool st[N];
void add(int a,int b,int c){
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
//以start为起点到其他牧场的距离
int dijkstra(int start){
memset(dist,0x3f,sizeof dist);
memset(st,0,sizeof st);
dist[start]=0;
priority_queue<PII,vector<PII>,greater<PII>> heap;
heap.push({0,start});
while(heap.size()){
auto t=heap.top();
heap.pop();
int d=t.first,ver=t.second;
if(st[ver]) continue;
st[ver]=true;
for(int i=h[ver];~i;i=ne[i]){
int j=e[i];
if(dist[j]>d+w[i]){
dist[j] =d+w[i];
heap.push({dist[j],j});
}
}
}
int t=0;
for(int i=0;i<p;i++) {
//牛分别在哪个牧场
int j=cow[i];
if(dist[j]==INF) return INF;
t+=dist[j];
}
return t;
}
int main(){
cin>>p>>n>>m;
memset(h,-1,sizeof h);
for(int i=0;i<p;i++) cin>>cow[i];
for(int i=0;i<m;i++){
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
int res=0x3f3f3f3f;
for(int i=1;i<=n;i++){
res=min(res,dijkstra(i));
}
cout<<res;
return 0;
}