洛谷P1828 香甜的黃油 Sweet Butter

香甜的黃油 Sweet Butter

黃油真的是這么做的嗎？！！！[惶恐]
這道題是Dijkstra算法的簡單變形
通過題意我們要找到一個點使奶牛所在點的路程和最短。通過Dijkstra的模板我們可以求的一點到其他任一點的最短路徑，那么我們可以遍歷所有點跑一邊Dijkstra，每個點的最后把奶牛所在點的dis相加，最后將每個點的路程和進行比較，輸出最短。
AC代碼如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#define INF 0x3f3f3f3f
#define MAXN 1000
using namespace std;vector<int> G[MAXN];
vector<int> M[MAXN];int cow[MAXN];
int dis[MAXN];
int done[MAXN];struct item
{int u;int dis;friend bool operator < (struct item x,struct item y){return x.dis>y.dis;}
};priority_queue<item> q;
int mn=INF;
int main()
{int n,p,c;scanf("%d%d%d",&n,&p,&c);for(int i=1;i<=n;i++){scanf("%d",&cow[i]);}for(int i=1;i<=c;i++){int x,y,z;scanf("%d%d%d",&x,&y,&z);G[x].push_back(y);M[x].push_back(z);G[y].push_back(x);M[y].push_back(z);}for(int i=1;i<=p;i++){memset(dis,0x3f,sizeof(dis));memset(done,0,sizeof(done));q.push((item){i,0});dis[i]=0;while(!q.empty()){item r;r=q.top();q.pop();int au=r.u;if(done[au]) continue;done[au]=1;for(int j=0;j<G[au].size();j++){int p=G[au][j];int c=dis[au]+M[au][j];if(dis[p]>c){dis[p]=c;q.push((item){p,c});}}}int total=0;for(int j=1;j<=n;j++){if(cow[j]==INF) continue;total=total+dis[cow[j]];}mn=min(mn,total);}printf("%d",mn);return 0;
}