http://codeforces.com/problemset/problem/986/A
做法:
对于每一种k,bfs每个镇上要取得的最短路径,最后对于每个镇根据k的值sort排序取s个输出即可。
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
#include<stack>
#include<map>
#include<queue>
#include<cstdio>
using namespace std;
#define mod 1000000007
#define INF 0x3f3f3f3f
typedef long long ll;
const int maxn=1e5+10;
vector<int> G[maxn];
int n,m,k,s,a[maxn],vis[maxn],dis[maxn][105];
queue<int> q;
void bfs()
{
for(int j=1;j<=k;j++)
{
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++)
if(a[i]==j)
q.push(i),dis[i][j]=0,vis[i]=1;
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(vis[v]==0) //该节点未访问
{
dis[v][j]=dis[u][j]+1;
vis[v]=1;
q.push(v);
}
}
}
}
}
int main()
{
int x,y;
while(cin>>n>>m>>k>>s)
{
for(int i=0;i<=n;i++)
G[i].clear();
for(int i=1;i<=n;i++)
cin>>a[i];
while(m--)
{
cin>>x>>y;
G[x].push_back(y);
G[y].push_back(x);
}
bfs();
for(int i=1;i<=n;i++)
{
sort(dis[i]+1,dis[i]+1+k);
int ans=0;
for(int j=1;j<=s;j++)
ans+=dis[i][j];
cout<<ans<<" ";
}
cout<<endl;
}
return 0;
}