2021百度之星 初赛第一场 T3 鸽子

前言

好久没做 dp 了，没想出来竟然是这么做的。

题面

题解

很妙啊

不过最后写的时候因为 HDU 奇葩的行末空格和文末回车机制 PE 了四五发…

#include <cstdio>
#include <algorithm>
using std :: min;

int read() {
	int ans = 0; char c = getchar();
	while('0' > c || c > '9') c = getchar();
	while('0' <= c && c <= '9') ans = (ans << 3) + (ans << 1) + (c ^ '0'), c = getchar();
	return ans;
}

const int maxn = 100000 + 6, inf = 0x7fffffff - maxn;
int f[maxn];

/// f[i][j] 表示前 i 次操作，让 电脑 k 在位置 j 的最小代价 
/// 由于 f[i] 与 f[i+1] 至多只有一个位置不同，因此可以减一维状态 

int main() {
	int T = read();
	while(T --) {
		int n, m, k; scanf("%d%d%d", &n, &m, &k);
		for(int i = 1; i <= n; i ++) f[i] = inf; /// 初始化 f[0][...] 
		f[k] = 0;
		for(int i = 1; i <= m; i ++) {
			int u, v; scanf("%d%d", &u, &v); /// 递推得到 f[i] 
			int lastu = f[u], lastv = f[v];
			f[v] = min(lastv + 1, lastu);
			f[u] = min(lastu + 1, lastv); /// 注意更新顺序 
		}
		printf("%d", f[1] >= inf ? -1 : f[1]);
		for(int i = 2; i <= n; i ++) printf(" %d", f[i] >= inf ? -1 : f[i]); /// 不允许有行末空格 
		putchar('\n'); /// 要求有文末回车 
	}
	return 0;
}