[数据结构与算法]1345：【例4-6】香甜的黄油

• 一本通传送门
• SPFA例题
• 由于SPFA求的是单源最短路，要枚举所有点，再用SPFA
• 顺便一提，为了SPFA简单点我用了链式前向星（是的我不会别的）

代码

#include<bits/stdc++.h>
using namespace std;
int n,p,c,x,y,z,cnt,a[1000],d[1000];
struct cyy{//链式前向星超好用
    int to,w,next;
}b[15000];
int h[15000];
void add(int x,int y,int z){//加边函数
	cnt++;
    b[cnt].to=y,b[cnt].w=z,b[cnt].next=h[x];
	h[x]=cnt;
}
queue<int> f;
bool s[1510];
void spfa(int q){//SPFA计算最短路
    memset(d,0x3f,sizeof(d));//记得初始化
    memset(s,0,sizeof(s));
    d[q]=0,s[q]=1;f.push(q);
    while(!f.empty()){
        int t=f.front();f.pop();s[t]=0;//出队
        for(int i=h[t];i;i=b[i].next)
            if(d[b[i].to]>d[t]+b[i].w){//更新
                d[b[i].to]=d[t]+b[i].w;
                if(!s[b[i].to]){//入队
                    s[b[i].to]=1;
                    f.push(b[i].to);
                }
            }
    }
}
int main()
{
    scanf("%d%d%d",&n,&p,&c);
    for(int i=1;i<=n;i++){//是i号牛在x号牧场，不是i号牧场有x头牛
        scanf("%d",&x);
        a[x]++;
    }
    for(int i=1;i<=c;i++){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);//道路是双向的，加边时注意
    }
    int ans=0x7fffffff;
    for(int i=1;i<=p;i++){
        spfa(i);
        int t=0;
        for(int j=1;j<=p;j++)
            t+=a[j]*d[j];//计算路程和，所以是本牧场的牛*两地路程
        ans=min(t,ans);//试试能不能更新
    }
    cout<<ans<<endl;
    return 0;
}

创作打卡挑战赛

赢取流量/现金/CSDN周边激励大奖