Codeforces Round #485(div 2) D

题目描述

链接

n: 城镇的数量

m: 道路的数量

k: 商品种类的数量

s: 举办展览会所需商品种类的最小值

a[i]: 城镇i拥有的商品种类

思路

对于任意商品(i), 使用广搜bfs求出送到任意城镇(j)所需的最小花费(dis[i][j])即可

求dis的时间复杂度为(O(km))

代码

#include <bits/stdc++.h>

using namespace std;
int n,m,k,s;
int cst[100005];
vector<int> v[100005]; // 邻接表保存稀疏图 
vector<int> gud[105];  // 拥有展品i的城市 
int dis[105][100005]; // dis[i][j]为把展品i运送到城市j的最小花费 
void bfs(vector<int> g,int* d) {
	//int ans=1;
	queue<int> q;
	fill(d+1,d+1+n,-1);
	for(int x:g) {
		q.push(x);
		d[x]=0;
	}
	while(!q.empty()) {
		int l=q.front();
		q.pop();
		for(int k:v[l]) {
			if(d[k]==-1) {
				d[k]=d[l]+1;
				q.push(k);
			}
		}
	}
}
int main() {
	cin>>n>>m>>k>>s;

	for(int i=1; i<=n; i++) {
		int bb;
		cin>>bb;
		gud[bb].push_back(i);
	}
	for(int i=1; i<=m; i++) {
		int a,b;
		cin>>a>>b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	for(int i=1; i<=k; i++) {
		bfs(gud[i],dis[i]);
	}
	for(int i=1; i<=n; i++) {
		vector<int> v;
		for(int j=1; j<=k; j++)
			v.push_back(dis[j][i]);
		sort(v.begin(),v.end());
		cout<<accumulate(v.begin(),v.begin()+s,0)<<" ";
	}
	return 0;
}