제출 #1335799

#제출 시각아이디문제언어결과실행 시간메모리
1335799i271828JJOOII 2 (JOI20_ho_t2)C++20
100 / 100
17 ms4284 KiB
#include <bits/stdc++.h>
#define ll long long
#define pii pair<int,int>
using namespace std;

const int MAX=2e5+5;
const int INF=1<<30;

int N,K;
int A[MAX];
int nxt[3][MAX];

int main(){
	cin>>N>>K;
	for (int i=0;i<N;i++){
		char c;cin>>c;
		if (c=='J') A[i]=0;
		else if (c=='O') A[i]=1;
		else A[i]=2;
	}
	for (int t=0;t<3;t++){
		fill_n(nxt[t],N+1,INF);
		vector<int> vct;
		for (int i=0;i<N;i++) if (A[i]==t) vct.push_back(i);
		for (int i=0;i<((int)vct.size())-(K-1);i++){
			nxt[t][vct[i]]=vct[i+(K-1)];
		}
		for (int i=N-1;i>=0;i--) nxt[t][i]=min(nxt[t][i],nxt[t][i+1]);
	}
	int ans=INF;
	for (int i=0;i<N;i++){
		int j=i;
		for (int t=0;t<3;t++){
			j=nxt[t][j]+1;
			if (j>=INF) break;
		}
		if (j>=INF) continue;
		ans=min(ans,j-i-3*K);
	}
	if (ans>=INF) ans=-1;
	cout<<ans<<'\n';
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...