香甜的黄油

题目描述 Description

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

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

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

输入描述 Input Description

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

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

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

输出描述 Output Description

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

样例输入 Sample Input

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

样例图形

         P2  
P1 @--1--@ C1
    \    |\
     \   | \
      5  7  3
       \ |   \
        \|    \ C3
      C2 @--5--@
         P3    P4

样例输出 Sample Output

8

{说明： 放在4号牧场最优. }



题解：由于道路是双向的，因此只需要枚举每一个牧场，然后计算它到各个牧场所需的路程。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define M 810
using namespace std;
int n,p,c,ans=0x7ffffff;
int a[M];
int num[M],b[M][M],v[M][M],team[2*M],dis[M];
bool f[M];
void spfa(int k)
{
   for (int j=1;j<=p;j++) dis[j]=0x7ffffff; 
   memset(team,0,sizeof(team));
   memset(f,0,sizeof(f));
   int head(0),tail(1); 
   dis[k]=0;team[1]=k;f[k]=1;
   do
      {
          head++;
          head=((head-1)%1601)+1;//循环队列处理 
          int ki=team[head];
          f[ki]=0;
          for (int j=1;j<=num[ki];j++)
            if (dis[b[ki][j]]>dis[ki]+v[ki][b[ki][j]])
                {
                   dis[b[ki][j]]=dis[ki]+v[ki][b[ki][j]];
                   if (!f[b[ki][j]])
                      {
                          tail++;
                          tail=((tail-1)%1601)+1;
                          team[tail]=b[ki][j];
                          f[b[ki][j]]=1;      
                      }
                }
      }while (head!=tail);
} 
int main()
{
    scanf("%d%d%d",&n,&p,&c);
    for (int i=1;i<=p;i++)
      {
          a[i]=0;
          num[i]=0;
          for (int j=1;j<=p;j++) v[i][j]=0x7ffffff;
      }
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1,x,y,z;i<=c;i++)
      {
           scanf("%d%d%d",&x,&y,&z);
           v[x][y]=v[y][x]=z;
           b[x][++num[x]]=y;
           b[y][++num[y]]=x;
      }
    for (int i=1;i<=p;i++)
      {
                spfa(i);
              int sum(0);
              for (int j=1;j<=n;j++) sum+=dis[a[j]];
              ans=min(ans,sum);
      }
    cout<<ans<<endl;
    return 0;
      
}