AT4821 Yutori（貪心）

Description

給定一個長爲 $n$ 的字符串，第 $i$ 個字符爲 o 意味着第 $i$ 天可以工作，否則不可以。第 $i$ 天工作後的 $i + 1 \sim i + c$ 天都不能工作。如果工作 $k$ 天，求哪幾天必須工作。

$1 \leq n \leq 2 \times 10^5$。

Solution

先正向求第 $i$ 次工作最早在 $\text{pre}_i$ 這一天，再反向求第 $i$ 次工作最晚在 $\text{suf}_i$ 這一天。如果 $\text{pre}_i = \text{suf}_i$ 那麼第 $i$ 天必須工作。

時間複雜度 $O(n)$。

Code

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5, INF = 0x3f3f3f3f;
inline int read() {
	int x = 0, f = 0; char ch = 0;
	while (!isdigit(ch)) f |= ch == '-', ch = getchar();
	while (isdigit(ch)) x = (x << 3) + (x << 1) + (ch ^ 48), ch = getchar();
	return f ? -x : x;
}
int pre[N], suf[N];
char s[N]; 
int main() {
	int n = read(), k = read(), c = read();
	scanf("%s", s + 1); int len = strlen(s + 1);
	for (int i = 1, j = 0; i <= len && j <= k; i++) if (s[i] == 'o') pre[++j] = i, i += c;
	for (int i = len, j = k; i && j; i--) if (s[i] == 'o') suf[j--] = i, i -= c;
	for (int i = 1; i <= k; i++) if (pre[i] == suf[i] && pre[i]) printf("%d\n", pre[i]);
	return 0;
}