一本通 第三部分 数据结构 第四章 图论算法 第二节 最短路径算法 1345：【例4-6】香甜的黄油

1345：【例4-6】香甜的黄油

时间限制: 1000 ms 内存限制: 65536 KB
提交数: 1161 通过数: 541
【题目描述】
农夫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），当然，连接是双向的。

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

【输入样例】
3 4 5
2
3
4
1 2 1
1 3 5
2 3 7
2 4 3
3 4 5
【输出样例】
8
【提示】
说明：放在4号牧场最优。

#include<bits/stdc++.h>
using namespace std;
int g[1001][1001],dis[1001],a[1001][1001],b[1001],num[1001];
queue<int>q;
int main()
{
int n,m,p,minn=2147483647;
cin>>n>>p>>m;
memset(g,0x7f,sizeof(g));
for(int i=1;i<=n;i++)
cin>>b[i];
for(int i=1;i<=m;i++)
{
int x,y,z;
cin>>x>>y>>z;
g[x][y]=g[y][x]=z;
a[x][++num[x]]=y;
a[y][++num[y]]=x;
}
for(int i=1;i<=p;i++)
{
fill_n(dis,1002,999999);
q.push(i);
dis[i]=0;
while(!q.empty())
{
int u=q.front();
q.pop();
for(int j=1;j<=num[u];j++)
if(g[u][a[u][j]]+dis[u]<dis[a[u][j]])
{
dis[a[u][j]]=g[u][a[u][j]]+dis[u];
q.push(a[u][j]);
}
}
int tot=0;
for(int j=1;j<=n;j++)
tot+=dis[b[j]];
if(tot<minn)
minn=tot;
}
cout<<minn;
return 0;
}