AtCoder Beginner Contest 161 E - Yutori

AtCoder Beginner Contest 161 E - Yutori

题意

神仙\(LFX\)要在接下来的\(N\)天中工作\(K\)天，当他工作一天之后在接下来的\(C\)天都不能工作，给一个长度为\(N\)的字符串\(S\)，\(S[i]=\text' \mathit x\text '\)表示在第\(i\)天不可以工作，问有哪几天必须工作才能保证能工作\(K\)天。

分析

考虑去最大化工作日的天数的贪心过程，会得到一个长度为\(k\)的数组\(L\)，\(L[x]\)表示第\(x\)个工作日不能早于第\(L[x]\)天，

同理我们可以得到数组\(R\)，\(R[x]\)表示第\(x\)个工作日不能晚于第\(R[x]\)天。若存在一个\(x\)满足\(L[x]=R[x]=i\)，则第\(i\)天必须工作。

Code

#include<algorithm>
#include<iostream>
#include<cstring>
#include<iomanip>
#include<sstream>
#include<cstdio>
#include<string>
#include<vector>
#include<bitset>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define rep(i,x,n) for(int i=x;i<=n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
#define sz(a) int(a.size())
#define rson mid+1,r,p<<1|1
#define pii pair<int,int>
#define lson l,mid,p<<1
#define ll long long
#define pb push_back
#define mp make_pair
#define se second
#define fi first
using namespace std;
const double eps=1e-8;
const int mod=1e9+7;
const int N=2e5+10;
const int inf=1e9;
int n,k,c;
char s[N];
int L[N],R[N];
int main(){
	//ios::sync_with_stdio(false);
	//freopen("in","r",stdin);
	cin>>n>>k>>c;
	cin>>s+1;
	int pre=-inf,j=0;
	for(int i=1;i<=n;i++){
		if(s[i]=='o'&&i-pre-1>=c){
			pre=i;
			L[++j]=i;
		}
	}
	pre=inf,j=k+1;
	for(int i=n;i>=1;i--){
		if(s[i]=='o'&&pre-i-1>=c&&j>=1){
			pre=i;
			R[--j]=i;
		}		
	}
	for(int i=1;i<=k;i++){
		if(L[i]==R[i]) cout<<L[i]<<endl;
	}
	return 0;
}