Codeforces Round #485 (Div. 1) A. Fair（多源点BFS + 转换思维）

Codeforces Round #485 (Div. 1) A. Fair（多源点BFS + 转换思维）

题意：
$n$个点每个点都有一种商品，共有$k$种商品（从1到$k$编号），$m$条边（保证连通），求每个点为起点买到$s$种商品的最小距离
思路：
如果以每个点为起点进行$n$次$bfs$肯定超时，想到以每种商品进行$bfs$，求出每种商品到第$i$个点的最小距离。因为只有$k$种商品，只需要进行$k$次$bfs$。最后对每个点到$k$种商品的距离排序，取前$s$个即可

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const long long mod=1e9+7; 
const ll inf=1e9+7;
const int maxn=4e5+5;
int n,m,k,s,r,ans,to[maxn],nex[maxn],head[maxn],w[maxn],dis[maxn][105];//每种商品到 每个点的最小距离 
void add(int x,int y){
	to[++r]=y;
	nex[r]=head[x];
	head[x]=r;
}
void bfs(int now){
    queue<int>p;
	for(int i=1;i<=n;i++){
		dis[i][now]=inf;
		if(w[i]==now){
			p.push(i);
			dis[i][now]=0;
		}
	}
	while(!p.empty()){
		int x=p.front();p.pop();
		for(int i=head[x];i;i=nex[i]){
			int y=to[i];
			if(dis[y][now]>dis[x][now]+1){
				p.push(y);
				dis[y][now]=dis[x][now]+1;
			}
		}
	}
}
int main(){ 
    scanf("%d%d%d%d",&n,&m,&k,&s); 
    for(int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    int x,y;
    for(int i=1;i<=m;i++){
    	scanf("%d%d",&x,&y);
    	add(x,y),add(y,x);
	}
    for(int i=1;i<=k;i++)bfs(i);
    for(int i=1;i<=n;i++){
    	ans=0;
    	sort(dis[i]+1,dis[i]+1+k);
    	for(int j=1;j<=s;j++)
    		ans+=dis[i][j];
    	printf("%d ",ans);
	}
}