莫欺少年穷，修魔之旅在这开始—>算法提高课题解

思路：

1. 核心：求解每个牧场到牛所在牧场最短路之和的最小值
2. 用 spfa 求解一个牧场到其他牧场的最短路，如果是牛所在牧场就用 res 加上

#include<bits/stdc++.h>

using namespace std;

const int N = 810, M = 3000, INF = 0x3f3f3f3f;

int n,p,m;
int id[N];
int h[N],e[M],w[M],ne[M],idx;
int dist[N];
bool st[N];

void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

int spfa(int start)
{
    memset(dist,0x3f,sizeof dist);
    dist[start]=0;

    queue<int> q;
    q.push(start);
    st[start]=true;

    //spfa求解这个牧场到其他牧场的最短路
    while(q.size())
    {
        auto t=q.front();
        q.pop();
        st[t]=false;

        for(int i=h[t];~i;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[t]+w[i])
            {
                dist[j]=dist[t]+w[i];
                if(!st[j])
                {
                    q.push(j);
                    st[j]=true;
                }
            }
        }
    }

    //这个牧场到牛所在牧场但距离之和
    int res=0;
    for(int i=0;i<n;i++)
        if(dist[id[i]]==INF) return INF;
        else res+=dist[id[i]];

    return res;
}

int main()
{
    cin>>n>>p>>m;

    for(int i=0;i<n;i++) cin>>id[i];

    memset(h,-1,sizeof h);

    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }

    //枚举每个牧场到牛所在牧场距离之和的最小值
    int res=INF;
    for(int i=1;i<=p;i++) res=min(res,spfa(i));

    cout<<res<<endl;

    return 0;
}