SPFA - 香甜的黄油 Sweet Butter - 洛谷 P1828

SPFA - 香甜的黄油 Sweet Butter - 洛谷 P1828

农夫John发现了做出全威斯康辛州最甜的黄油的方法：糖。

把糖放在一片牧场上，他知道 N 只奶牛会过来舔它，这样就能做出能卖好价钱的超甜黄油。

当然，他将付出额外的费用在奶牛上。

农夫John很狡猾，就像以前的巴甫洛夫，他知道他可以训练这些奶牛，让它们在听到铃声时去一个特定的牧场。

他打算将糖放在那里然后下午发出铃声，以至他可以在晚上挤奶。

农夫John知道每只奶牛都在各自喜欢的牧场（一个牧场不一定只有一头牛）。

给出各头牛在的牧场和牧场间的路线，找出使所有牛到达的路程和最短的牧场（他将把糖放在那）。

数据保证至少存在一个牧场和所有牛所在的牧场连通。

输入格式

第一行: 三个数：奶牛数 N，牧场数 P，牧场间道路数 C。

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

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

输出格式

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

数据范围

$1\le N\le 500,2\le P\le 800,1\le C\le 1450,1\le D\le 255$

输入样例：

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

输出样例：

8

---

分析：

$题意：$

$给定一个P个点，C条边的无向图，另从P个点中选一个点S，$

$使得给定的N个点到点S的最短距离之和最小。$

$可见，本题是一个多源最短路问题。$

$考虑到floyed算法的时间复杂度不够优秀，$

$可以跑N次spfa算法或堆优化的dijkstra算法。$

$这里跑N次spfa算法，每次返回距离之和。$

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>

using namespace std;

const int N=810,M=3000,inf=0x3f3f3f3f;

int k,n,m,e[M],ne[M],h[N],w[M],idx;
int dis[N],S[510];
bool st[N];

void add(int a,int b,int c)
{
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}

int spfa(int start)
{
    memset(dis,0x3f,sizeof dis);
    memset(st,false,sizeof st);
    
    dis[start]=0;
    
    queue<int> Q;
    Q.push(start);
    st[start]=true;
    
    while(Q.size())
    {
        int u=Q.front();
        Q.pop();
        st[u]=false;
        
        for(int i=h[u];~i;i=ne[i])
        {
            int j=e[i];
            if(dis[j]>dis[u]+w[i])
            {
                dis[j]=dis[u]+w[i];
                if(!st[j])
                {
                    st[j]=true;
                    Q.push(j);
                }
            }
        }
    }
    
    int s=0;
    for(int i=1;i<=k;i++)
        if(dis[S[i]]==inf) return inf;
        else s+=dis[S[i]];
    
    return s;
}

int main()
{
    memset(h,-1,sizeof h);
    
    scanf("%d%d%d",&k,&n,&m);
    for(int i=1;i<=k;i++) scanf("%d",&S[i]);
    
    int a,b,c;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
    
    int res=inf;
    for(int i=1;i<=n;i++) res=min(res,spfa(i));
    
    cout<<res<<endl;
    
    return 0;
}