c++香甜的黄油

香甜的黄油

描述

农夫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），当然，连接是双向的。

输出

一行输出奶牛必须行走的最小的距离和.

输入样例 1 

3 4 5
2
3
4
1 2 1
1 3 5
2 3 7
2 4 3
3 4 5

输出样例 1

8

#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;

const int MAXN = 800 + 10;
const int MAXP = 500 + 10;
const int INF = 0x3f3f3f3f;

int N, P, C;
int cow_at[MAXP];
vector<pair<int, int>> graph[MAXN];
vector<int> dijkstra(int src) {
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
    vector<int> dist(P + 1, INF);
    vector<bool> visited(P + 1, false);
    pq.push(make_pair(0, src));
    dist[src] = 0;

    while (!pq.empty()) {
        int u = pq.top().second;
        pq.pop();

        if (visited[u]) continue;

        visited[u] = true;

        for (const auto &adj : graph[u]) {
            int v = adj.first;
            int weight = adj.second;
            if (dist[u] + weight < dist[v]) {
                dist[v] = dist[u] + weight;
                pq.push(make_pair(dist[v], v));
            }
        }
    }
    return dist;
}

int main() {
    cin >> N >> P >> C;
    for (int i = 0; i < N; ++i) {
        cin >> cow_at[i];
    }

    for (int i = 0; i < C; ++i) {
        int a, b, d;
        cin >> a >> b >> d;
        graph[a].push_back(make_pair(b, d));
        graph[b].push_back(make_pair(a, d));
    }

    int min_distance_sum = INF;
    for (int i = 1; i <= P; ++i) {
        vector<int> dist = dijkstra(i);
        int distance_sum = 0;
        for (int j = 0; j < N; ++j) {
            distance_sum += dist[cow_at[j]];
        }
        min_distance_sum = min(min_distance_sum, distance_sum);
    }

    cout << min_distance_sum << endl;
    return 0;
}