解題思路：

1. 對物品i bfs，更新每個小鎮j獲得每個物品i的最短距離。

2. 時間復雜度o(n* k*logk)，滿足2s的要求。

代碼：

#include <iostream> #include <queue> #include <list> #include <algorithm> #include <stdio.h> #include <string.h> using namespace std; typedef long long ll; #define MAX 2000000000 int a[500050]; list <int> e[500050]; int d[500050][110]; bool used[500010]; queue <int> q;int main(){ // fill(d[0],d[0]+500050*110,MAX);int n,m,k,s;scanf("%d%d%d%d", &n, &m, &k, &s);for(int i = 1;i <= n; ++i){scanf("%d", &a[i]);}int u,v;for(int i = 1;i <= m; ++i){scanf("%d%d", &u, &v);e[u].push_back(v);e[v].push_back(u);}for(int i = 1;i <= k; ++i){memset(used,false,sizeof(used));for(int j = 1;j <= n; ++j){if(a[j] == i){d[j][i] = 0;q.push(j);used[j] = true;}}while(q.size()){int x = q.front();for(auto l:e[x]){if(!used[l]){d[l][i] = d[x][i]+1;used[l] = true;q.push(l);}}q.pop();}}for(int i = 1;i <= n; ++i){sort(d[i]+1,d[i]+1+k);int ans = 0;for(int j = 1;j <= s; ++j){ans += d[i][j];}printf("%d ",ans);}printf("\n");return 0; }

總結

以上是生活随笔為你收集整理的Codeforces 986A. Fair（对物品bfs暴力求解）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。