acwing 1127 香甜的黄油(单源最短路)

题面

题解

代码

#include<bits/stdc++.h>

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


int n,p,m;
int h[N],e[M],ne[M],w[M],idx;
int id[N];
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;
	while(q.size()){
		auto t = q.front();
		q.pop();
		st[t]=false;
		for(int i=h[t];i!=-1;i=ne[i]){
			int j=e[i];
			if(dist[j]>dist[t]+w[i]){
				dist[j]=dist[t]+w[i];
				if(!st[j]){
					st[j]=true;
					q.push(j);
				}
			}
		}
	}
	int res=0;
	for(int i=0;i<n;i++){
		int j=id[i];
		if(dist[j]==INF) return INF;
		else res+=dist[j];
	}
	return res;
}

int main(){
	memset(h,-1,sizeof h);
	cin>>n>>p>>m;
	
	for(int i=0;i<n;i++) cin>>id[i];
	
	for(int i=0;i<m;i++){
		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;
}