1211: 小镇购物

题目描述

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");
    }
}