【USACO 3.2.6】香甜的黄油

【描述】

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

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

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

【格式】

INPUT FORMAT:

第一行: 三个数：奶牛数N，牧场数P（2<=P<=800），牧场间道路数C(1<=C<=1450)

第二行到第N+1行: 1到N头奶牛所在的牧场号

第N+2行到第N+C+1行： 每行有三个数：相连的牧场A、B，两牧场间距离D（1<=D<=255），当然,连接是双向的

OUTPUT FORMAT:

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

【分析】

直接上dijkstral+堆就可以了。

 #include <cstdlib>
 #include <iostream>
 #include <cstring>
 #include <cmath>
 #include <cstdio>
 #include <queue>
 #include <stack>
 #include <ctime>
 #include <set>
 #include <vector>
 const int maxp=+;
 const int INF=;
 using namespace std;
 struct edge{int to,w;};
 struct node
 {
        int num;//编号
        int dist;//距离
        bool operator <(const node &b)const
        {
             return dist<b.dist;
        }
 };
 vector<edge>map[maxp];
 int cow[maxp];
 int n,p,c,low[maxp];
 int ans=INF;

 void add_Edge(int u,int v,int w);
 void dijkstra(int sta);

 int main()
 {
     int i,j;
     //文件操作
     freopen("data.txt","r",stdin);
     freopen("out.txt","w",stdout);
     memset(cow,,sizeof(cow));
     scanf("%d%d%d",&n,&p,&c);

     for (i=;i<=n;i++)
     {
         int u;
         scanf("%d",&u);
         cow[u]++;//放置奶牛
     }
     for (i=;i<=c;i++)
     {
         int u,v,w;
         scanf("%d%d%d",&u,&v,&w);
         add_Edge(u,v,w);//无向边
         add_Edge(v,u,w);
     }
     for (i=;i<=p;i++) dijkstra(i);//单源最短路
     printf("%d",ans);
     return ;
 }
 void dijkstra(int sta)
 {
      int i,j,cnt=,vis[maxp];
      priority_queue<pair<int,int> >heap;
      for (i=;i<=p;i++)
      {
          low[i]=INF;
          vis[i]=;
      }//初始化
      heap.push(make_pair(, sta));
      low[sta]=;
      while (!heap.empty())
      {
         int u=heap.top().second;
         heap.pop();
         if (!vis[u])
         {
             vis[u]=true;
             for (int i=; i <map[u].size(); i++)
             {
                 int v=map[u][i].to,cost;
                 if (!vis[v]) cost=map[u][i].w;
                 if (low[v]>low[u]+cost)
                 {
                     low[v]=low[u]+cost;
                     heap.push(make_pair(-low[v], v));
                 }
             }
         }
     }
      //printf("%d:\n",sta);
      //for (i=1;i<=p;i++) printf("%d ",low[i]);
      //printf("\n");
      for (i=;i<=p;i++) cnt=cnt+(low[i]*cow[i]);
      ans=min(ans,cnt);
 }
 void add_Edge(int u,int v,int w)
 {
      map[u].push_back((edge){v,w});
 }