【例4-6】香甜的黄油

【例4-6】香甜的黄油

链接：http://ybt.ssoier.cn:8088/problem_show.php?pid=1345
时间限制: 1000 ms         内存限制: 65536 KB

 

 

【题目描述】

农夫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<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#define For(a,b,c) for(int a=b;a<=c;a++)
using namespace std;
const int maxn = 1500 ;

struct edge{
    int to,co;
    edge(int tt,int cc){
        to=tt;co=cc;
    }
    edge(){}
};
vector<edge>G[maxn];
bool inq[maxn];
int dis[maxn],c[maxn];
int N,P,C;

int  SPFA(int k)
{
    memset(dis,127,sizeof(dis));
    int ans=0;
    queue<int> Q;
    dis[k]=0;
    Q.push(k);
    inq[k]=1;
    
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop();
        inq[u]=0;
        
        for(int i=0;i<G[u].size();i++)
        {
            edge &e=G[u][i];
            if(dis[e.to]>dis[u]+e.co)
            {
                dis[e.to]=dis[u]+e.co;
                if(!inq[e.to])
                {
                    Q.push(e.to);
                    inq[e.to]=1;
                }
            }
        }
    }
    For(i,1,N)
        ans+=dis[c[i]];
    return ans;
}
int main()
{
    int minn = 100000000;
    cin>>N>>P>>C;
    For(i,1,N)scanf("%d",&c[i]);
    
    For(i,1,C)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        G[u].push_back(edge(v,w));
        G[v].push_back(edge(u,w));        
    }
    For(i,1,P)
        minn=min(minn,SPFA(i));
    cout<<minn<<endl;
}