洛谷 P1828 【香甜的黄油 Sweet Butter】

这道题应该就是模板题了吧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;
}