Farmer John 发现了做出全威斯康辛州最甜的黄油的方法:糖。
把糖放在一片牧场上,他知道 N NN 只奶牛会过来舔它,这样就能做出能卖好价钱的超甜黄油。当然,他将付出额外的费用在奶牛上。
Farmer John 很狡猾。像以前的 Pavlov,他知道他可以训练这些奶牛,让它们在听到铃声时去一个特定的牧场。他打算将糖放在那里然后下午发出铃声,以至他可以在晚上挤奶。
Farmer John 知道每只奶牛都在各自喜欢的牧场(一个牧场不一定只有一头牛)。给出各头牛在的牧场和牧场间的路线,找出使所有牛到达的路程和最短的牧场(他将把糖放在那)。
输入格式
第一行包含三个整数 N , P , C,分别表示奶牛数、牧场数和牧场间道路数。
第二行到第 N+1 行,每行一个整数,其中第 i 行的整数表示第 i−1 头奶牛所在的牧场号。
第 N+2 行到第 N+C+1 行,每行包含三个整数 A,B,D,表示牧场号为 A 和 B 的两个牧场之间有一条长度为 D 的双向道路相连。
输出格式
输出一行一个整数,表示奶牛必须行走的最小的距离和。
这道题实际上就是要求出所有有牛的牧场的到每个牧场的最短路。由于牧场的数量比较多,这里不能使用 floyd 算法了,所以可以考虑使用 p 次 SPFA。
然后剩下的工作就是枚举每个牧场然后把每一头奶牛到这个牧场的最短路径长度加起来求个和,比较所有的和的最小值。
#include <bits/stdc++.h>
using namespace std;
const int N = 1001;
const int M = 10101;
const int inf = 0x3f3f3f3f;
struct Edge {
int to, next, len;
} e[M * 2];
int h[N], tot = -1, n, p, c, d[N], inq[N], cnt[N], x, ans = 1e9;
queue<int> q;
void addEdge(int x, int y, int len) {
e[++tot] = {y, h[x], len};
h[x] = tot;
}
void spfa(int s) {
memset(d, 0x3f, sizeof(d));
memset(inq, 0, sizeof(inq));
d[s] = 0;
inq[s] = true;
q.push(s);
while (!q.empty()) {
int v = q.front();
q.pop();
inq[v] = false;
for (int i = h[v]; i != -1; i = e[i].next) {
int to = e[i].to;
if (d[to] > d[v] + e[i].len) {
d[to] = d[v] + e[i].len;
if (!inq[to]) {
q.push(to);
inq[to] = true;
}
}
}
}
int sum = 0;
for (int i = 1; i <= p; i++) {
sum += cnt[i] * d[i];
}
ans = min(ans, sum);
}
int main() {
cin >> n >> p >> c;
memset(h, -1, sizeof(h));
for (int i = 1; i <= n; i++) {
cin >> x;
cnt[x]++;
}
for (int i = 0; i < c; i++) {
int u, v, w;
cin >> u >> v >> w;
addEdge(u, v, w);
addEdge(v, u, w);
}
for (int i = 1; i <= p; i++) {
spfa(i);
}
cout << ans;
return 0;
}