D. Fair (bfs)

题目链接:D. Fair

题意：

给出n个村庄，m条路，共有k种特产，对每个村庄都需要带s种特产。问每个村庄最小的花费。

思路：看完题目感觉是bfs，可是不大会写，刚开始的思路是想对每个城市bfs找出离最近的三个特产，但是发现会t。看了题解，特产总数有限 对特产进行bfs，算出每个特产到能到达城市的最小距离,d[i][j] 表示第i个城市带第j个特产花费，然后sort一下，前s个就是最小花费和。

小技巧：

bfs的时候将相同特产的城市推入，vis表示距离，vis初始化为-1，-1表示为没搜索过。
代码：

#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int M=1e5+5;
const int INF=0x3f3f3f3f;
vector<int>V[M];
int n,m,k,s,tp[M];
int vis[M],d[M][105];
void bfs(int x)
{
    ios::sync_with_stdio(false);
    memset(vis,-1,sizeof(vis));
    queue<int>q;
    for(int i=1;i<=n;++i)
    {
        if(tp[i]==x)
        {
            q.push(i);
            vis[i]=0;
        }
    }
    while(!q.empty())
    {
        int p=q.front();
        q.pop();
        for(int i=0;i<V[p].size();++i)
        {
            if(vis[V[p][i]]==-1)
            {
                vis[V[p][i]]=vis[p]+1;
                q.push(V[p][i]);
            }
        }
    }
    for(int i=1;i<=n;++i)
    if(vis[i]!=-1){
    d[i][x]=vis[i];
     }
}
int main()
{
    memset(d, INF, sizeof(d));  
    int i,j;
   cin>>n>>m>>k>>s;
   for(i=1;i<=n;++i)
   cin>>tp[i];
   for(i=1;i<=m;++i)
   {
    int u,v;
    cin>>u>>v;
    V[u].push_back(v);
    V[v].push_back(u);
   }
   for(i=1;i<=k;++i)
   bfs(i);

   for(i=1;i<=n;++i)
   {
    sort(d[i]+1,d[i]+k+1);
    int sum=0;
    for(j=1;j<=s;++j)
    sum+=d[i][j];
    cout<<sum<<" ";
   }
}