Submission #964555

#	Time	Username	Problem	Language	Result	Execution time	Memory
964555		onebit1024	JJOOII 2 (JOI20_ho_t2)	C++17	100 / 100	9 ms	3440 KiB

ho_t2

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

#define int long long
#define pb push_back
#define all(c) c.begin(), c.end()
#define endl "\n"

const double PI=3.141592653589;


void __print(int x) {cerr << x;}
void __print(long x) {cerr << x;}
void __print(unsigned x) {cerr << x;}
void __print(unsigned long x) {cerr << x;}
void __print(unsigned long long x) {cerr << x;}
void __print(float x) {cerr << x;}
void __print(double x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '\"' << x << '\"';}
void __print(const string &x) {cerr << '\"' << x << '\"';}
void __print(bool x) {cerr << (x ? "true" : "false");}

template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ','; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? "," : ""), __print(i); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}
#ifndef ONLINE_JUDGE
#define dbg(x...) cerr << "LINE(" << __LINE__ << ") -> " <<"[" << #x << "] = ["; _print(x)
#else
#define dbg(x...)
#endif



void solve()
{
    int n,k;cin >> n >> k;
    string s;cin >> s;
    s = "0"+s;
    vector<int>J,O,I;
    for(int i =1 ;i<=n;++i){
        if(s[i] == 'J')J.pb(i);
        else if(s[i]=='O')O.pb(i);
        else I.pb(i);
    }
    int res = 1e18;
    for(int i = 0;i<((J.size())-k+1);++i){
        int l = J[i+k-1];
        int c = upper_bound(all(O),l)-O.begin();
        if((c+k-1) < O.size()){
            l = O[c+k-1];
            c = upper_bound(all(I),l)-I.begin();
            if((c+k-1) < I.size()){
                l = I[c+k-1];
                res = min(res, l-J[i]+1 - (k*3));
            }
        }
    }
    if(res == 1e18)res =-1;
    cout << res << endl;
}   

int32_t main()
{

    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);


    // freopen("input.txt", "r", stdin);
    // freopen("output.txt", "w", stdout);
    

    int T=1;
    for(int i = 1;i<=T;++i)
    {
        // cout << "Case #" << i << ": ";
        solve();
    }
}

Compilation message (stderr)

ho_t2.cpp: In function 'void solve()':
ho_t2.cpp:52:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'long long unsigned int' [-Wsign-compare]
   52 |     for(int i = 0;i<((J.size())-k+1);++i){
      |                   ~^~~~~~~~~~~~~~~~~
ho_t2.cpp:55:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |         if((c+k-1) < O.size()){
      |            ~~~~~~~~^~~~~~~~~~
ho_t2.cpp:58:24: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   58 |             if((c+k-1) < I.size()){
      |                ~~~~~~~~^~~~~~~~~~

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