oj.uz

Submission #509620

# Submission time Handle Problem Language Result Execution time Memory
509620 2022-01-14T07:29:30 Z Khizri JJOOII 2 (JOI20_ho_t2) C++17
100 / 100
 9 ms 3932 KB 
#include <bits/stdc++.h>
using namespace std;
//------------------------------DEFINE------------------------------
//******************************************************************
#define IOS ios_base::sync_with_stdio(false); cin.tie(0),cout.tie(0)
#define ll long long
#define pb push_back
#define F first
#define S second
#define INF 1e18
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define pii pair<int,int>
#define pll pair<ll,ll>
#define OK cout<<"Ok"<<endl;
#define MOD (ll)(1e9+7)
#define endl "\n"
//******************************************************************
//----------------------------FUNCTION------------------------------
//******************************************************************
ll gcd(ll a,ll b){
    if(a>b) swap(a,b);
    if(a==0) return a+b;
    return gcd(b%a,a);
}
ll lcm(ll a,ll b){
    return a/gcd(a,b)*b;
}
bool is_prime(ll n){
    ll k=sqrt(n);
    if(n==2) return true;
    if(n<2||n%2==0||k*k==n) return false;
    for(int i=3;i<=k;i+=2){
        if(n%i==0){
            return false;
        }
    }
    return true;
}
//*****************************************************************
//--------------------------MAIN-CODE------------------------------
const int mxn=2e5+5;
int t=1,n,k,pref[mxn][4];
string s;
int f(char c){
    if(c=='J') return 0;
    if(c=='O') return 1;
    return 2;
}
void solve(){
    cin>>n>>k>>s;
    int a=-1,b=-1,x=-1,y=-1,l=-1,r=-1,ans=1e6,say=0;
    pref[0][f(s[0])]=1;
    for(int i=1;i<s.size();i++){
        pref[i][0]=pref[i-1][0];
        pref[i][1]=pref[i-1][1];
        pref[i][2]=pref[i-1][2];
        pref[i][f(s[i])]=pref[i-1][f(s[i])]+1;
    }
    for(int i=0;i<n&&say<k;i++){
        if(s[i]=='J'){
            say++;
            if(a==-1){
                a=i;
            }
            if(say==k){
                b=i;
                break;
            }
        }
    }
    if(say<k){
        cout<<-1<<endl;
        return;
    }
    say=0;
    for(int i=b+1;i<n&&say<k;i++){
        if(s[i]=='O'){
            say++;
            if(x==-1){
                x=i;
            }
            if(say==k){
                y=i;
                break;
            }
        }
    }
    if(say<k){
        cout<<-1<<endl;
        return;
    }
    say=0;
    for(int i=y+1;i<n&&say<k;i++){
        if(s[i]=='I'){
            say++;
            if(l==-1){
                l=i;
            }
            if(say==k){
                r=i;
                break;
            }
        }
    }
    if(say<k){
        cout<<-1<<endl;
        return;
    }
    ans=min(ans,r-a+1-k*3);
    while(true){
        do{
            a++;
        }while(s[a]!='J'&&a<n);
        if(a==n) break;
        b=max(b,a);
        bool q=true;
        for(int i=b;i<n;i++){
            if(pref[i][0]-pref[a-1][0]>=k){
                b=i;
                q=false;
                break;
            }
        }
        if(q){
            break;
        }
        x=max(b+1,x);
        y=max(x,y);
        q=true;
        for(int i=y;i<n;i++){
            if(pref[i][1]-pref[x-1][1]>=k){
                y=i;
                q=false;
                break;
            }
        }
        if(q){
            break;
        }
        l=max(l,y+1);
        r=max(r,l);
        q=true;
        for(int i=r;i<n;i++){
            if(pref[i][2]-pref[l-1][2]>=k){
                r=i;
                q=false;
                break;
            }
        }
        if(q){
            break;
        }
        ans=min(ans,r-a+1-k*3);
    }
    if(ans==1e6){
        cout<<-1<<endl;
        return;
    }
    cout<<ans<<endl;
}
int main(){
    IOS;
    //cin>>t;
    while(t--){
        solve();
    }
    return 0;
}

Compilation message

ho_t2.cpp: In function 'void solve()':
ho_t2.cpp:54:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   54 |     for(int i=1;i<s.size();i++){
      |                 ~^~~~~~~~~

Subtask #1
1.0 / 1.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 316 KB Output is correct
3 Correct 0 ms 324 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 320 KB Output is correct
9 Correct 1 ms 204 KB Output is correct
10 Correct 1 ms 204 KB Output is correct
11 Correct 1 ms 316 KB Output is correct
12 Correct 1 ms 204 KB Output is correct
13 Correct 0 ms 324 KB Output is correct
14 Correct 0 ms 204 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 316 KB Output is correct
3 Correct 0 ms 324 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 320 KB Output is correct
9 Correct 1 ms 204 KB Output is correct
10 Correct 1 ms 204 KB Output is correct
11 Correct 1 ms 316 KB Output is correct
12 Correct 1 ms 204 KB Output is correct
13 Correct 0 ms 324 KB Output is correct
14 Correct 0 ms 204 KB Output is correct
15 Correct 1 ms 332 KB Output is correct
16 Correct 1 ms 332 KB Output is correct
17 Correct 1 ms 324 KB Output is correct
18 Correct 0 ms 332 KB Output is correct
19 Correct 0 ms 332 KB Output is correct
20 Correct 1 ms 316 KB Output is correct
21 Correct 1 ms 332 KB Output is correct
22 Correct 1 ms 272 KB Output is correct
23 Correct 0 ms 332 KB Output is correct
24 Correct 0 ms 332 KB Output is correct
25 Correct 1 ms 320 KB Output is correct
26 Correct 1 ms 332 KB Output is correct
27 Correct 1 ms 332 KB Output is correct
28 Correct 1 ms 332 KB Output is correct
29 Correct 1 ms 332 KB Output is correct
30 Correct 1 ms 324 KB Output is correct
31 Correct 1 ms 332 KB Output is correct
32 Correct 1 ms 320 KB Output is correct
33 Correct 1 ms 320 KB Output is correct
34 Correct 1 ms 332 KB Output is correct
35 Correct 1 ms 332 KB Output is correct

Subtask #3
87.0 / 87.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 316 KB Output is correct
3 Correct 0 ms 324 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 320 KB Output is correct
9 Correct 1 ms 204 KB Output is correct
10 Correct 1 ms 204 KB Output is correct
11 Correct 1 ms 316 KB Output is correct
12 Correct 1 ms 204 KB Output is correct
13 Correct 0 ms 324 KB Output is correct
14 Correct 0 ms 204 KB Output is correct
15 Correct 1 ms 332 KB Output is correct
16 Correct 1 ms 332 KB Output is correct
17 Correct 1 ms 324 KB Output is correct
18 Correct 0 ms 332 KB Output is correct
19 Correct 0 ms 332 KB Output is correct
20 Correct 1 ms 316 KB Output is correct
21 Correct 1 ms 332 KB Output is correct
22 Correct 1 ms 272 KB Output is correct
23 Correct 0 ms 332 KB Output is correct
24 Correct 0 ms 332 KB Output is correct
25 Correct 1 ms 320 KB Output is correct
26 Correct 1 ms 332 KB Output is correct
27 Correct 1 ms 332 KB Output is correct
28 Correct 1 ms 332 KB Output is correct
29 Correct 1 ms 332 KB Output is correct
30 Correct 1 ms 324 KB Output is correct
31 Correct 1 ms 332 KB Output is correct
32 Correct 1 ms 320 KB Output is correct
33 Correct 1 ms 320 KB Output is correct
34 Correct 1 ms 332 KB Output is correct
35 Correct 1 ms 332 KB Output is correct
36 Correct 5 ms 3660 KB Output is correct
37 Correct 7 ms 3856 KB Output is correct
38 Correct 6 ms 3928 KB Output is correct
39 Correct 8 ms 3820 KB Output is correct
40 Correct 5 ms 3864 KB Output is correct
41 Correct 8 ms 3928 KB Output is correct
42 Correct 9 ms 3928 KB Output is correct
43 Correct 5 ms 2580 KB Output is correct
44 Correct 5 ms 3112 KB Output is correct
45 Correct 6 ms 3928 KB Output is correct
46 Correct 5 ms 3892 KB Output is correct
47 Correct 6 ms 3924 KB Output is correct
48 Correct 5 ms 3928 KB Output is correct
49 Correct 3 ms 2776 KB Output is correct
50 Correct 5 ms 3928 KB Output is correct
51 Correct 4 ms 3908 KB Output is correct
52 Correct 4 ms 3812 KB Output is correct
53 Correct 5 ms 3924 KB Output is correct
54 Correct 4 ms 3928 KB Output is correct
55 Correct 5 ms 3932 KB Output is correct
56 Correct 4 ms 3888 KB Output is correct