Submission #717233

#	Time	Username	Problem	Language	Result	Execution time	Memory
717233		vjudge1	JJOOII 2 (JOI20_ho_t2)	C++17	100 / 100	11 ms	2048 KiB

ho_t2

#include <bits/stdc++.h>
using namespace std;

int n,k,ans = INT_MAX;
string s;
vector<int> a,b,c;

int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0);

    cin >> n >> k >> s;
    for(int i = 0;i < n;i++)
    {
        char ch = s[i];
        if(ch=='J') a.push_back(i);
        else if(ch=='O') b.push_back(i);
        else c.push_back(i);
    }
    for(int i = 0;i+k-1 < a.size();i++)
    {
        int x = a[i+k-1];
        auto ib = upper_bound(b.begin(),b.end(),x);
        if((int)(b.end()-ib)<k) break;
        ib+=k-1;
        x = *ib;
        auto ic = upper_bound(c.begin(),c.end(),x);
        if((int)(c.end()-ic)<k) break;
        ic+=k-1;
        ans = min(ans,*ic-a[i]+1-(k*3));
    }
    if(ans==INT_MAX) ans = -1;
    cout << ans;
}

Compilation message (stderr)

ho_t2.cpp: In function 'int main()':
ho_t2.cpp:20:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |     for(int i = 0;i+k-1 < a.size();i++)
      |                   ~~~~~~^~~~~~~~~~

• Subtask 1: 1 / 1
• Subtask 2: 12 / 12
• Subtask 3: 87 / 87