【最短路】香甜的黄油

原题链接

枚举所有的牧场，每次算出所有点到此牧场的最短路，再累加所有奶牛喜欢的牧场的最短路，比较得出最小值。数据规模对spfa比较友善。

#include<bits/stdc++.h>
#define N 500+10
#define M 800+10
using namespace std;

int ans=1e9;
int c,n,m;
int cow[N],d[M],used[M];
vector<int>v[M];
vector<int>val[M];

void spfa(int st){
    memset(d,0x3f,sizeof d);
    memset(used,0,sizeof used);
    queue<int>q;int cnt=0;
    
    q.push(st);d[st]=0;used[st]=1;
    
    while(q.size()){
        int x=q.front();q.pop();
        used[x]=0;
        
        for(int i=0;i<v[x].size();i++){
            int y=v[x][i],z=val[x][i];
            
            if(d[y]>d[x]+z){
                d[y]=d[x]+z;
                
                if(!used[y]){
                    q.push(y);
                    used[y]=1;
                }
            }
        }
    }
    
    for(int i=1;i<=c;i++)
        cnt+=d[cow[i]];
    
    ans=min(ans,cnt);
}

int main(){
    cin>>c>>n>>m;
    
    for(int i=1;i<=c;i++)
        cin>>cow[i];
        
    for(int i=1;i<=m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        if(a==b)continue;
        
        v[a].push_back(b);
        val[a].push_back(c);
        v[b].push_back(a);
        val[b].push_back(c);
    }
        
    for(int i=1;i<=n;i++)
        spfa(i);
        
    cout<<ans;
    return 0;
}