题解：CF928B Chat

对于数列

a_{1},a_{2}…a_{n}

，对于每个

a_{i}

：

$a_{i} = 0$ 答案为第 $i$ 个左右 $k$ 个，注意判断是否可以取。
$a_{i} \ne 0$ 观察到题目保证 $a_{i} ≤ i$ ，所以答案为第 $i$ 个左右 $k$ 个加上第 $a_{i}$ 个的答案减去重叠。

对于不同的情况，我们只要分开判断就行了。

代码

CPP

#include <bits/stdc++.h>
using namespace std;
int a[100005];
int main() {
	int n, k;
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		int m;
		cin >> m;
		if (m == 0) {
			a[i] = min(k, i - 1) + 1;
		} else {
			a[i] = a[m] + min(k * 2 + 1, i - m);
		}
		cout << a[i] + min(k, n - i) <<" ";
	}
	return 0;
}

文章操作

代码

相关推荐

评论

题解：CF928B Chat