Fair CodeForces - 987D (bfs)

链接

大意:给定无向图边权均为1, 每个节点有一种货物, 对于每个节点, 求出拿到$s$种不同货物的最短距离 (每种货物独立计算,并且不用返回)

因为$s$较小, 直接枚举每种货物即可

所以问题就转化为无向图中标记了一个点集, 对于每个点, 求它到这个点集的最短距离

这跟昨天做的980E比较像, 不过980E是树上的情况, 并且是动态添点的

但该题点集是固定不变的, 直接把所有点扔进队列里BFS就行了

复杂度$O((n+m)k)$

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <queue>
#define pb push_back
#define REP(i,a,n) for(int i=a;i<=n;++i)
using namespace std;

const int N = 4e5+10, INF = 0x3f3f3f3f;
int n, m, k, s;
vector<int> g[N];
queue<int> q[111];
int d[N][111], vis[N][111];

int main() {
	scanf("%d%d%d%d", &n, &m, &k, &s);
	REP(i,1,n) {
		int t;
		scanf("%d", &t);
		q[t].push(i);
		vis[i][t] = 1;
	}
	REP(i,1,m) {
		int x, y;
		scanf("%d%d", &x, &y);
		g[x].pb(y),g[y].pb(x);
	}
	REP(i,1,k) {
		while (!q[i].empty()) {
			int x = q[i].front();q[i].pop();
			for (int y:g[x]) if (!vis[y][i]) {
				d[y][i] = d[x][i]+1;
				vis[y][i] = 1;
				q[i].push(y);
			}
		}
	}
	REP(i,1,n) {
		sort(d[i]+1,d[i]+1+k);
		int ans = 0;
		REP(j,1,s) ans += d[i][j];
		printf("%d ", ans);
	} puts("");
}