这道题应该就是模板题了吧qwq。
统计每一个牧场的放糖的情况,选择最优的即可 有亿点水的绿题。
#include <bits/stdc++.h>
using namespace std;
int n , p , c , ans = 0x3ffffff , sum;
int vis[810] , dis[810] , f[810];
vector<pair<int , int> > e[810]; //vector香~
void work(int s){ //正常的最短路
//经过上次调一道题调很久,最后发现是卡SPFA后,就不敢随随便便写SPFA了qnq
priority_queue<pair<int , int> > q;
memset(dis , 127 , sizeof(dis));
memset(f , 0 , sizeof(f));
dis[s] = 0;
q.push(make_pair(0 , s));
while(!q.empty()){
int x = q.top().second;
q.pop();
if(f[x]) continue;
f[x] = 1;
for(int i = 0; i < e[x].size(); i++){
int nx = e[x][i].first , w = e[x][i].second;
if(dis[nx] > dis[x] + w){
dis[nx] = dis[x] + w;
q.push(make_pair(-dis[nx] , nx));
}
}
}
}
int main(){
cin >> n >> p >> c;
for(int i = 1; i <= n; i++){
int x;
cin >> x;
vis[x]++; //统计这个牧场的奶牛数
}
for(int i = 1; i <= c; i++){
int x , y , z;
cin >> x >> y >> z;
e[x].push_back(make_pair(y , z));
e[y].push_back(make_pair(x , z));
}
for(int i = 1; i <= p; i++){
sum = 0;
work(i);
for(int j = 1; j <= p; j++) sum += (vis[j] * dis[j]); //一个个枚举
ans = min(ans , sum);
}
cout << ans;
return 0;
}