Codeforces 986 A

传送门

标签是数论，结果是个最短路，有点点小懵逼。

BFS

题目大意

n个城市，每个城市分别产生一种商品，商品种类一共有k中，问从第i个城市出发至少得到s件商品的最短路径。

思路

对于每一个小镇i，求出到每件物品j的最短距离即可。

代码

int n,m,s,k;
int a[maxn];
vector <int> e[maxn];
int vis[maxn];
int step[maxn][120];


int main(){
	cin>>n>>m>>k>>s;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	int v,u;
	for(int i=1;i<=m;i++){
		cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	
	for(int i=1;i<=k;i++){
		queue<int>q;
		memset(vis,0,sizeof vis);
		for(int j=1;j<=n;j++){
			if(a[j]==i){
				step[j][i]=0;
				q.push(j);
				vis[j]=1;
			}
		}
		while(!q.empty()){
			int x=q.front();
			q.pop();
			for(int j=0;j<e[x].size();j++){
				if(!vis[e[x][j]]){
					step[e[x][j]][i]=step[x][i]+1;
					vis[e[x][j]]=1;
					q.push(e[x][j]);
				}
			}
		}
	}
	for(int i=1;i<=n;i++){
		sort(step[i]+1,step[i]+1+k);
		int ans=0;
		for(int j=1;j<=s;j++){
			ans+=step[i][j];
		}
		printf("%d ",ans);
	}
	printf("\n");
}