【USACO3.2.6】香甜的黄油

题目描述

农夫John发现做出全威斯康辛州最甜的黄油的方法——糖。把糖放在一片牧场上，他知道 N（1<=N<=500）只奶牛会过来舔它，这样就能做出能卖好价钱的超甜黄油。当然，他将付出额外的费用在奶牛上。 

农夫John很狡猾。像以前的巴甫洛夫，他知道他可以训练这些奶牛，让它们在听到铃声时去一个特定的牧场。他打算将糖放在那里然后下午发出铃声，以至他可以在晚上挤奶。

农夫John知道每只奶牛都在各自喜欢的牧场（一个牧场不一定只有一头牛）。给出各头牛在的牧场和牧场间的路线，找出使所有牛到达的路程和最短的牧场（他将把糖放在那）。 

输入格式

第一行: 三个数：奶牛数 N，牧场数 P（2<=P<=800），牧场间道路数 C(1<=C<=1450)。
第二行到第 N+1 行: 1到 N 头奶牛所在的牧场号。
第 N+2 行到第 N+C+1 行：每行有三个数：相连的牧场 A、B，两牧场间距（1<=D<=255），当然，连接是双向的。

输出格式

一行输出奶牛必须行走的最小的距离和。

样例数据 1

输入

3 4 5
2
3
4
1 2 1
1 3 5
2 3 7
2 4 3
3 4 5

输出

8

备注

【样例说明】
把糖放在4号牧场最优，奶牛行走的距离和最小。

 

解析：
       最短路。

       PS：做水题有益身心健康。

 

代码：
 

#include <bits/stdc++.h>
using namespace std;

const int Maxn=805;
const int Maxm=1500;
int tot,n,m,ans=1e9,size,sum;
int first[Maxn],dis[Maxn],id[Maxn],vis[Maxn];
struct shu{int to,next,len;};
shu edge[Maxm<<1];

inline void build(int x,int y,int z)
{
	edge[++size].next=first[x];
	first[x]=size;
	edge[size].to=y,edge[size].len=z;
}

inline void solve(int s)
{
	sum=0,memset(dis,0x33f3f,sizeof(dis)),memset(vis,0,sizeof(vis));
	priority_queue<pair<int,int> >q;
	q.push(make_pair(0,s)),dis[s]=0;
	while(q.size())
	{
	  int point=q.top().second;q.pop();
	  if(vis[point]) continue;vis[point]=1;
	  for(int u=first[point];u;u=edge[u].next)
	  {
	   	int to=edge[u].to;
	   	if(dis[to] > dis[point]+edge[u].len)
	   	{
	   	  dis[to]=dis[point]+edge[u].len;
	   	  q.push(make_pair(-dis[to],to));
	   	}
	  }
	}
	for(int i=1;i<=tot;i++) sum+=dis[id[i]];
	ans=min(ans,sum);
}

int main()
{
	scanf("%d%d%d",&tot,&n,&m);
	for(int i=1;i<=tot;i++) scanf("%d",&id[i]);
	for(int i=1;i<=m;i++)
	{
	  int x,y,z;
	  scanf("%d%d%d",&x,&y,&z);
	  build(x,y,z),build(y,x,z);
	}
	for(int i=1;i<=n;i++) solve(i);
	cout<<ans;
	return 0;
}