南沙C++信奥赛陈老师解一本通题 1345：【例4-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），当然，连接是双向的。
【输出】
一行 输出奶牛必须行走的最小的距离和。
【输入样例】
3 4 5
2
3
4
1 2 1
1 3 5
2 3 7
2 4 3
3 4 5【输出样例】
8#include <bits/stdc++.h>
using namespace std;
int g[1001][1001];
int d[1001][1001];
int a[1001]; //代表每个牧场有的奶牛数量
#define INF 0x3f3f3f3f
void Floyd(int n)
{
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(d[i][j]>d[i][k]+d[k][j])
d[i][j]=d[i][k]+d[k][j];
}
int main()
{
int n,p,c,ans=INF,cow;
cin>>n>>p>>c;
for(int i=1;i<=n;i++)
cin>>a[i];
memset(g,INF,sizeof(g));
memset(d,INF,sizeof(d));
for(int i=1;i<=c;i++)
{
int x,y,w;
cin>>x>>y>>w;
g[x][y]=g[y][x]=w;
d[x][y]=d[y][x]=w;
}
Floyd(p);
for(int i=1;i<=p;i++) //假设放在任何牧场编号i
{
int total=0;
for(int j=1;j<=n;j++)
if(a[j]!=0 &&i!=a[j] ) //代表有路径 自己不需要路径 可以有多头牛，这里要多次运算
total+=d[ a[j] ][i];
ans=min(ans,total);
}
cout<<ans;
return 0;
}