#include <bits/stdc++.h>
using namespace std;
vector<int> sumki_J(string napis,int k,int n){
vector<int>wynik(1),sumy(1);
int suma=0,index=-1;
bool czybyl=false;
queue<int>NIENAWIDZETEGOGOWNA;
for(int i=0;i<n;++i){
if(napis[i]=='J') {
if(index==-1) {
index=i;
NIENAWIDZETEGOGOWNA.push(i);
}
suma++;
}
else{
if(suma>0)
sumy.push_back(sumy[i]+1);
else
sumy.push_back(sumy[i]);
}
if(suma==k && napis[i]=='J'){
czybyl=true;
sumy.push_back(sumy[i]-sumy[NIENAWIDZETEGOGOWNA.front()+1]);
index=i;
NIENAWIDZETEGOGOWNA.push(i);
NIENAWIDZETEGOGOWNA.pop();
suma--;
}
else if(napis[i]=='J')
sumy.push_back(sumy[i]);
if(czybyl==true)
wynik.push_back(sumy[i+1]);
else
wynik.push_back(-1);
}
return wynik;
}
vector<int> sumki_I(string napis,int k,int n){
vector<int>wynik(1),sumy(1);
int suma=0,index=-1;
bool czybyl=false;
queue<int>jebac;
for(int i=n-1;i>=0;--i){
if(napis[i]=='I') {
if(index==-1) {
index=n-i-1;
jebac.push(index);
}
suma++;
}
else{
if(suma>0)
sumy.push_back(sumy[n-i-1]+1);
else
sumy.push_back(sumy[n-i-1]);
}
if(suma==k && napis[i]=='I'){
czybyl=true;
sumy.push_back(sumy[n-i-1]-sumy[jebac.front()+1]);
index=n-i-1;
jebac.push(index);
jebac.pop();
suma--;
}
else if(napis[i]=='I')
sumy.push_back(sumy[n-i-1]);
if(czybyl==true)
wynik.push_back(sumy[n-i]);
else
wynik.push_back(-1);
}
return wynik;
}
int main()
{
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int n,k,s=0;
string napis;
cin >> n >> k;
cin >> napis;
vector<int>sum_I=sumki_I(napis,k,n);
vector<int>sum_J=sumki_J(napis,k,n);
vector<int>sum_O(1);
queue<int>pozycje;
for(int i=0;i<n;++i){
if(napis[i]=='O')
sum_O.push_back(sum_O[i]);
else
sum_O.push_back(sum_O[i]+1);
}
int wynik=1000000;
int s1,s2,gowno;
reverse(sum_I.begin(),sum_I.end());
for(int i=0;i<n;++i){
if(napis[i]=='O'){
s++;
pozycje.push(i);
}
if(s==k){
s--;
s2=sum_J[pozycje.front()];
gowno=sum_O[i+1]-sum_O[pozycje.front()];
s1=sum_I[pozycje.front()+2+gowno];
//cout << s1 << ' ' << s2 << endl;
if(s1!=-1 && s2!=-1)
wynik=min(wynik,s1+s2+gowno);
pozycje.pop();
}
}
if(wynik!=1000000)
cout << wynik;
else
cout << -1;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
336 KB |
Output is correct |
2 |
Incorrect |
1 ms |
504 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
336 KB |
Output is correct |
2 |
Incorrect |
1 ms |
504 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
336 KB |
Output is correct |
2 |
Incorrect |
1 ms |
504 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |