Submission #1114670

#	Time	Username	Problem	Language	Result	Execution time	Memory
1114670		AdamGS	JJOOII 2 (JOI20_ho_t2)	C++17	100 / 100	12 ms	2644 KiB

ho_t2

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n,k;
    cin>>n>>k;
    string s;
    cin>>s;
    vector<int> j;
    vector<int> l;
    for(int i = 0;i<n;i++)
    {
        if(s[i] == 'J')
        {
            j.push_back(i);
        }
        if(s[i] == 'O')
        {
            if(j.size() < k)
            {
                l.push_back(-1);
            }
            else
            {
                l.push_back(j[j.size() - k]);
            }
        }
    }
    vector<int> i;
    vector<int> r;
    for(int q = n-1;q>=0;q--)
    {
        if(s[q] == 'I')
        {
            i.push_back(q);
        }
        if(s[q] == 'O')
        {
            if(i.size()<k)
            {
                r.push_back(-1);
            }
            else
            {
                r.push_back(i[i.size()-k]);
            }
        }
    }
    reverse(r.begin(),r.end());
    int ans = -1;
    for(int q = 0;q<r.size()-k+1;q++)
    {
        if(l[q] == -1) continue;
        if(r[q+k-1] == -1) continue;
        ans = max(ans,l[q]+n-r[q+k-1]-1);
    }
    if(ans == -1) cout<<ans;
    else cout<<n-ans-k*3;
}

Compilation message (stderr)

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

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