codevs 2038 香甜的黄油 SPFA 解题报告

题目描述 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号牧场最优. }

数据范围及提示 Data Size & Hint

见描述

思路

很简单的最短路，枚举一下就好了。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int n,p,c;
int dis[801],a[801][801],b[501],dl[805],t,w,tot,mn=0x7fffffff;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void spfa(int x)
{
memset(dis,127,sizeof(dis));
int t=0,w=1;
bool pd[801]={0};
dl[t]=x;
pd[x]=1;
dis[x]=0;
while(t!=w)
{
for(int i=1;i<=p;i++)
{
if(a[dl[t]][i]+dis[dl[t]]<dis[i])
{
dis[i]=a[dl[t]][i]+dis[dl[t]];
if(!pd[i])
{
dl[w]=i;
w++;
w=(w-1)%801+1;
pd[i]=1;
}
}
}
pd[dl[t]]=0;
t++;
t=(t-1)%801+1;
}
tot=0;
for(int i=1;i<=n;i++)
tot+=dis[b[i]];
}
int main()
{
memset(a,127,sizeof(a));
n=read();p=read();c=read();
int x,y,z;
for(int i=1;i<=n;i++)
b[i]=read();
for (int i=1;i<=c;i++)
{
x=read();y=read();z=read();
a[x][y]=a[y][x]=z;
}
for(int i=1;i<=p;i++)
{
spfa(i);
if (tot<mn) mn=tot;
}
cout<<mn;
return 0;
}