洛谷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;
}