香甜的黄油 C++题解

香甜的黄油

内存限制： 256 MiB 时间限制： 1000 ms 标准输入输出 题目类型： 传统 评测方式： 文本比较

题目描述

农夫 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

#include <bits/stdc++.h>
#define int long long
#define PII pair<int, int>
using namespace std;
const int N = 1.5e3 + 5, INF = 0x3f3f3f3f3f3f3f3f;
int n, p, c, ans;
int Min = INF;
int id[N], dist[N];
vector<PII> a[N]; 
priority_queue<PII, vector<PII>, greater<PII> > Q;
int dijkstra(int s) {
	memset(dist, INF, sizeof(dist));
	ans = 0;
	dist[s] = 0;
	Q.push({0, s});
	while(!Q.empty()) {
		auto t = Q.top();
		Q.pop();
		int u = t.second;
		for(auto i : a[u]) {
			int v = i.second, w = i.first;
			if(dist[v] > dist[u] + w) { 
				dist[v] = dist[u] + w; 
				Q.push({dist[v], v}); 
			} 
		} 
	} 
	for(int j, i = 1; i <= n; i++) {
		j = id[i];
		ans += dist[j];
	} 
	return ans;
} 
signed main() {
	scanf("%lld %lld %lld", &n, &p, &c);
	for(int i = 1; i <= n; i++) {
		scanf("%lld", &id[i]);
	} 
	for(int x, y, z, i = 1; i <= c; i++) {
		scanf("%lld %lld %lld", &x, &y, &z);
		a[x].push_back({z, y});
		a[y].push_back({z, x});
	} 
	for(int i = 1; i <= p; i++) {
		Min = min(Min, dijkstra(i));
	} 
	printf("%lld", Min);
	return 0;
}