CF 986A Fair（多源BFS）

题目描述

一些公司将在Byteland举办商品交易会（or博览会？）。在Byteland有 nnn 个城市，城市间有 mmm 条双向道路。当然，城镇之间两两连通。 Byteland生产的货物有 kkk 种类型，每个城镇只生产一种。 为了举办商品交易会，你必须至少带来 sss 种不同类型的商品。将货物从 uuu 镇带到城镇 vvv 将花费 d(u,v)d(u,v)d(u,v) 的费用，其中 d(u,v)d(u,v)d(u,v) 是从 uuu 到 vvv 的最短路径的长度。 路径的长度是这个路径中的道路的数量。 组织者将支付所有的运输费用，但他们可以选择从哪些城镇带来货物。现在他们想计算每个城镇举办商品交易会的最小费用。

题解

我们考虑每种货物分开做。把生产同一种货物的城市扔到队列里跑BFS，得到每一种货物到每个点的最短距离。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
const int N=100010;
const int K=200;
int n,m,k,s,c[N],vis[N][K],head[N],cnt,ans;
queue<int> q; 
struct edge{
    int to,nxt;
}e[N*3];
void add(int u,int v){
    cnt++;
    e[cnt].nxt=head[u];
    e[cnt].to=v;
    head[u]=cnt;
}
int main(){
    scanf("%d%d%d%d",&n,&m,&k,&s);
    for(int i=1;i<=n;i++){
        scanf("%d",&c[i]); 
    }
    for(int i=1,u,v;i<=m;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    memset(vis,-1,sizeof(vis));
    for(int i=1;i<=k;i++){
        for(int j=1;j<=n;j++){
            if(c[j]==i)q.push(j),vis[j][i]=0;
        }
        while(!q.empty()){
            int u=q.front();
            q.pop();
            for(int x=head[u];x;x=e[x].nxt){
                int v=e[x].to;
                if(vis[v][i]>-1)continue;
                vis[v][i]=vis[u][i]+1;
                q.push(v);
            }
        }
    }
    for(int i=1;i<=n;i++){
        ans=0;
        sort(vis[i]+1,vis[i]+1+k);
        for(int j=1;j<=s;j++){
            ans+=vis[i][j];
        }
        printf("%d ",ans);
    }
    return 0;
}