【题目描述】
农夫John发现做出全威斯康辛州最甜的黄油的方法:糖。把糖放在一片牧场上,他知道N(1≤N≤500)只奶牛会过来舔它,这样就能做出能卖好价钱的超甜黄油。当然,他将付出额外的费用在奶牛上。
农夫John很狡猾。像以前的巴甫洛夫,他知道他可以训练这些奶牛,让它们在听到铃声时去一个特定的牧场。他打算将糖放在那里然后下午发出铃声,以至他可以在晚上挤奶。
农夫John知道每只奶牛都在各自喜欢的牧场(一个牧场不一定只有一头牛)。给出各头牛在的牧场和牧场间的路线,找出使所有牛到达的路程和最短的牧场(他将把糖放在那)。
【输入】
第一行: 三个数:奶牛数N,牧场数P(2≤P≤800),牧场间道路数C(1≤C≤1450)。
第二行到第N+1行: 1到N头奶牛所在的牧场号。
第N+2行到第N+C+1行:每行有三个数:相连的牧场A、B,两牧场间距(1≤D≤255),当然,连接是双向的。
【输出】
一行 输出奶牛必须行走的最小的距离和。
【输入样例】
3 4 5
2
3
4
1 2 1
1 3 5
2 3 7
2 4 3
3 4 5
【输出样例】
8
【提示】
说明:放在4号牧场最优。
#include<bits/stdc++.h>
using namespace std;
const int N=800+5,M=100000;
const int INF=0x3f3f3f3f;
int n,p,m,f[N];
struct edge{int to,w;};
struct node{
int id,dis;
bool operator<(const node& a)const{
return a.dis<dis;
}
};
vector<edge>e[N];
int Dijkstra(int s){
int dis[N],done[N];
priority_queue<node>q;
for(int i=1;i<=p;i++) dis[i]=INF,done[i]=0;
q.push({s,0});
dis[s]=0;
while(q.size()){
node u=q.top();
q.pop();
if(done[u.id]) continue;
done[u.id]=1;
for(int i=0;i<e[u.id].size();i++){
edge x=e[u.id][i];
if(done[x.to]) continue;
if(dis[x.to]>u.dis+x.w){
dis[x.to]=u.dis+x.w;
q.push({x.to,dis[x.to]});
}
}
}
int sum=0;
for(int i=1;i<=n;i++) if(s!=f[i]) sum+=dis[f[i]];
return sum;
}
int main(){
scanf("%d%d%d",&n,&p,&m);
for(int i=1;i<=n;i++) scanf("%d",&f[i]);
for(int i=0;i<m;i++){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
e[a].push_back({b,c});
e[b].push_back({a,c});
}
int res=INF;
for(int i=1;i<=p;i++){
// cout<<Dijkstra(i)<<endl;
res=min(res,Dijkstra(i));
}
cout<<res;
return 0;
}