Codeforces Round #485 (Div. 1) A. Fair(多源点BFS + 转换思维)
题意:
n
n
n个点每个点都有一种商品,共有
k
k
k种商品(从1到
k
k
k编号),
m
m
m条边(保证连通),求每个点为起点买到
s
s
s种商品的最小距离
思路:
如果以每个点为起点进行
n
n
n次
b
f
s
bfs
bfs肯定超时,想到以每种商品进行
b
f
s
bfs
bfs,求出每种商品到第
i
i
i个点的最小距离。因为只有
k
k
k种商品,只需要进行
k
k
k次
b
f
s
bfs
bfs。最后对每个点到
k
k
k种商品的距离排序,取前
s
s
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);
}
}