原题链接:https://www.luogu.org/problem/show?pid=1828#sub
经典的最短路问题。
各位不要被题目条件迷惑了,牧场想象成点,道路想象成边,奶牛所在的位置想象成点权就好。
输入的是无向图,所以在正向连边时反向连边。然后我们枚举所有的牧场,以枚举到的牧场作为起点寻求最短路,记录好dis数组。
算出最短路径后我们再枚举所有奶牛,开一个变量nowans记录以当前牧场为集合地点的所有牛需要走的最短路径,便有nowans += dis[c[j]] (j枚举自1到奶牛数)。
然后维护一下ans,输出就好了。
其实这题如果数据量较小的话可以用floyed求最短路。
参考代码:
#include <iostream>
#include <cstring>
#include <queue>
#include <cstdio>
#define maxn 23333
#define INF 2147483647
using namespace std;
struct Edge{
int from,to,dis;
};
Edge edge[maxn];
int head[maxn];
int dis[maxn];
int u,v,d,tot = ;
bool inq[maxn];
int c[maxn];
int N,P,C;
int ans = INF;
int nowans;
void add_edge(int from,int to,int dis){
edge[++tot].from = head[from];
edge[tot].to = to;
edge[tot].dis = dis;
head[from] = tot;
} void spfa(int s){
memset(inq,false,sizeof(inq));
for (int i=;i<=P;i++)
dis[i] = INF;
dis[s] = ;
queue<int> q;
q.push(s);
inq[s] = true;
while (!q.empty()){
int u = q.front();
q.pop();
inq[u] = false;
for (int i = head[u];i!=;i=edge[i].from){
int v = edge[i].to;
int w = edge[i].dis;
if (dis[u] + w < dis[v]){
dis[v] = w + dis[u];
if (!inq[v]){
inq[v] = true;
q.push(v);
}
}
}
}
} int main(){
cin >> N >> P >> C;
for (int i=;i<=N;i++)
cin >> c[i];
for (int i=;i<=C;i++){
cin >> u >> v >> d;
add_edge(u,v,d);
add_edge(v,u,d);
}
for (int i=;i<=P;i++){
spfa(i);
nowans = ;
for (int j=;j<=N;j++)
nowans += dis[c[j]];
ans = min(ans,nowans);
}
cout << ans << endl;
return ;
}