Sweet Butter 香甜的黄油

Description

农夫John发现做出全威斯康辛州最甜的黄油的方法：糖。把糖放在一片牧场上，他知道N（1<=N<=500）只奶牛会过来舔它，这样就能做出能卖好价钱的超甜黄油。当然，他将付出额外的费用在奶牛上。 农夫John很狡猾。像以前的Pavlov，他知道他可以训练这些奶牛，让它们在听到铃声时去一个特定的牧场。他打算将糖放在那里然后下午发出铃声，以至他可以在晚上挤奶。 农夫John知道每只奶牛都在各自喜欢的牧场（一个牧场不一定只有一头牛）。给出各头牛在的牧场和牧场间的路线，找出使所有牛到达的路程和最短的牧场（他将把糖放在那）
Input

第一行: 三个数：奶牛数N，牧场数（2<=P<=800），牧场间道路数C(1<=C<=1450) 第二行到第N+1行: 1到N头奶牛所在的牧场号 第N+2行到第N+C+1行： 每行有三个数：相连的牧场A、B，两牧场间距离D（1<=D<=255），当然,连接是双向的
Output

一行 输出奶牛必须行走的最小的距离和
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号牧场最优
}/

spfa:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
#include <utility>
#include <queue>
#include <map>
#include <string>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
int n,p,road_num;
typedef struct
{
    int to,w,next;
}node;
node edge[1450*2+10]; ///边上限1450，双向*2
int head[800+10],dis[800+10],cnt;
int cow_field[800+10];
bool vis[800+10];///牧场上限800

void add_edge(int u,int v,int w)
{
    edge[cnt].to = v;
    edge[cnt].w = w;
    edge[cnt].next = head[u];
    head[u] = cnt++;
}///链式前向星存图
void spfa(int pos)
{
    memset(vis,false,sizeof(vis));
    memset(dis,INF,sizeof(dis));///每次应该初始化
    queue<int> q;
    dis[pos] = 0;
    q.push(pos);
    vis[pos] = true;
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        vis[u] = false;
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int to = edge[i].to;
            if(dis[u]+edge[i].w<dis[to])
            {
                dis[to]=dis[u]+edge[i].w;
                if(!vis[to])
                {
                    q.push(to);
                    vis[to] = true;
                }
            }
        }
    }
}///spfa不解释
void input_and_add_edage()
{
    scanf("%d %d %d",&n,&p,&road_num);
    int t;///设一个临时变量，来存牛所在的牧场
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&t);
        cow_field[t]++;///此牧场的牛加1
    }
    int a,b,c;///临时存储起点，终点，距离
    memset(head,-1,sizeof(head));///初始化为-1
    for(int i=1;i<=road_num;i++)///链式前向星存图
    {
        scanf("%d %d %d",&a,&b,&c);
        add_edge(a,b,c);
        add_edge(b,a,c);///双向
    }
}
ll find_ans()
{
    ll ans_minn = 0x3f3f3f3f3f3f3f3fLL;///记得写LL
    for(int i=1;i<=p;i++)
    {
        spfa(i);///寻找从i点到每个点的最小路径值
        ll t = 0;
        for(int j=1;j<=p;j++)
        {
            t += (ll)cow_field[j]*dis[j];///表示j这个牧场的牛的个数 * j到i的距离（跑了spfa，已经是最小值）
        }
        if(t<ans_minn)
            ans_minn = t;///比较
    }
    return ans_minn;
}
int main()
{
    input_and_add_edage();///输入和存图
    ll ans = find_ans();///寻找答案
    printf("%lld\n",ans);
    return 0;
}

这是一份没过的：
floyd：显示 答案错误，现在还是没有弄明白，至少时间超限心里还舒服点

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
#include <utility>
#include <queue>
#include <map>
#include <string>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
 
int cow[500+5];
int dis[800+5][800+5];
int n,p,c;
void init()
{
    for(int i=1;i<=p;i++)
        for(int j=1;j<=p;j++)
            if(i==j) dis[i][j]=dis[j][i] = 0;
            else     dis[i][j]=dis[j][i] = INF;
}
void input()
{
    scanf("%d%d%d",&n,&p,&c);
    init();
    int t;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&t);
        cow[t]++;
    }
    int a,b,cc;
    for(int i=1;i<=c;i++)
    {
        scanf("%d%d%d",&a,&b,&cc);
        if( cc<dis[a][b] )
            dis[a][b]=dis[b][a]=cc;
    }
}
void floyd()
{
    for(int i=1;i<=p;i++)
        for(int j=1;j<=p;j++)
            for(int k=1;k<i;k++)
                if( dis[i][j]+dis[j][k]<dis[i][k]  )
                    dis[i][k] = dis[k][i] = dis[i][j]+dis[j][k];
}
int find_ans()
{
    int ans = INF;
    for(int i=1;i<=p;i++)
    {
        int sum = 0;
        for(int j=1;j<=p;j++)
            sum += cow[j]*dis[i][j];
        if(sum<ans) ans = sum;
    }
    return ans;
}
int main()
{
    input();
    floyd();
    printf("%d\n",find_ans());
    return 0;
}