题目大意:
给定一张$k$个结点,$m$条边的无向图,其中有$n$个点被标记,在这$k$个点中找出一个点使得这个点到那$n$个点的最短距离之和最小,求出这个距离和。
思路:
对于每个标记结点跑最短路,最后枚举每个结点,求出其到各个标记结点的最短距离和,取$min$。
#include<cstdio>
#include<cctype>
#include<functional>
#include<ext/pb_ds/priority_queue.hpp>
inline int getint() {
char ch;
while(!isdigit(ch=getchar()));
int x=ch^'';
while(isdigit(ch=getchar())) x=(((x<<)+x)<<)+(ch^'');
return x;
}
const int inf=0x7fffffff;
const int V=;
struct Edge {
int to,w;
};
std::vector<Edge> e[V];
inline void add_edge(const int u,const int v,const int w) {
e[u].push_back((Edge){v,w});
}
struct Vertex {
int id,dis;
bool operator > (const Vertex &another) const {
return dis>another.dis;
}
};
int k;
int dis[V][V];
__gnu_pbds::priority_queue<Vertex,std::greater<Vertex> > q;
__gnu_pbds::priority_queue<Vertex,std::greater<Vertex> >::point_iterator p[V];
inline void Dijkstra(const int s,int *dis) {
q.clear();
for(int i=;i<=k;i++) {
p[i]=q.push((Vertex){i,dis[i]=(i==s)?:inf});
}
while(q.top().dis!=inf) {
int x=q.top().id;
for(unsigned i=;i<e[x].size();i++) {
Edge &y=e[x][i];
if(dis[x]+y.w<dis[y.to]) {
q.modify(p[y.to],(Vertex){y.to,dis[y.to]=dis[x]+y.w});
}
}
q.modify(p[x],(Vertex){x,inf});
}
}
int main() {
int n=getint();
k=getint();
int c=getint();
int cow[n];
for(int i=;i<n;i++) cow[i]=getint();
while(c--) {
int u=getint(),v=getint(),w=getint();
add_edge(u,v,w);
add_edge(v,u,w);
}
for(int i=;i<n;i++) {
Dijkstra(cow[i],dis[i]);
}
int ans=inf;
for(int i=;i<=k;i++) {
int tmp=;
for(int j=;j<n;j++) {
tmp+=dis[j][i];
}
ans=std::min(ans,tmp);
}
printf("%d\n",ans);
return ;
}