题目描述
CSU镇上有n个商店,n个商店有m条双向小路相连,在这n个商店里共有k种不同商品,每个商店只有一种商品,每条路的权重都为1。现问你从每个商店出发,买够k种商品中的s种商品所需的最小代价,每个商店可以同时派出多个人买不同商品,买够即可。
输入
输入包含多组测试用例。
对于每一组输入包含四个数字n ,m, k,s (1<=n<=m<=1e5 , 1<=s<=k<=min(n,100))
分别代表商店数,小路数,商品种数,需要的商品数。
接下来n个数 a1,a2…an (1<=ai<=k),ai代表第i个商店的商品编号。
接下来m行小路(u,v),u≠v,代表商店u和v之间有小路连接。
输出
输出n个数字,第i个数字代表从商店i出发买够s种商品所需的最小代价。
样例输入
5 5 4 3
1 2 4 3 2
1 2
2 3
3 4
4 1
4 5
7 6 3 2
1 2 3 3 2 2 1
1 2
2 3
3 4
2 5
5 6
6 7
样例输出
2 2 2 2 3
1 1 1 2 2 1 1
#include<bits/stdc++.h>
using namespace std;
int n,m,k,s;
const int N=1e5+5;
vector<int> vec[N];///ans[i][j]代表第i个商店到第j件商品最近距离 既做标记数组又做计数数组
int ans[N][105];
int good[N];
void bfs(int a) ///多源bfs
{
queue<int>q;
for(int i=1;i<=n;i++)
{
if(good[i]==a) ///将同种商品全部入队
{
q.push(i);
ans[i][a]=0;
}
}
while(q.size())
{
int top=q.front();q.pop();
for(int i=0;i<vec[top].size();i++)
{
int v=vec[top][i];
if(ans[v][a]==-1)
{
ans[v][a]=ans[top][a]+1;
q.push(v);
}
}
}
}
main()
{
while(scanf("%d%d%d%d",&n,&m,&k,&s)!=EOF)
{
memset(ans,-1,sizeof(ans));
for(int i=1;i<=n;i++)
{
vec[i].clear();
scanf("%d",&good[i]);
}
for(int i=1;i<=m;i++)
{
int u,v;
scanf("%d %d",&u,&v);
vec[u].push_back(v);
vec[v].push_back(u);
}
for(int i=1;i<=k;i++)
bfs(i);
int res;
for(int i=1;i<=n;i++)
{
res=0;
sort(ans[i]+1,ans[i]+1+k);
for(int j=1;j<=s;j++)
{
res+=ans[i][j];
}
printf("%d ",res);
}
printf("\n");
}
}