Submission #1310058

#	Time	Username	Problem	Language	Result	Execution time	Memory
1310058		_Temirhan	JJOOII 2 (JOI20_ho_t2)	C++20	100 / 100	10 ms	3016 KiB

ho_t2

#include <bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>  
#include <ext/pb_ds/assoc_container.hpp>

using namespace std;
using namespace __gnu_pbds;

#if defined(__GNUC__) || defined(__clang__)
#pragma GCC optimize("Ofast,unroll-loops,inline-functions,no-stack-protector,03")
#pragma GCC target ("sse,sse2,sse3,ssse3,sse4,popcnt,abm,avx2,mmx,fma,avx,tune=native")
#pragma clang loop vectorize(enable)
#pragma clang loop interleave(enable)
#endif

#define int long long
#define sz(x) (int)x.size()
#define F first
#define S second
#define pb push_back
#define nl '\n'
#define o_set tree<int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update>
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
typedef pair<int, int> pii;

void file( string s = "" )
{
    if( s.empty() )
        return;
    freopen((s + ".in").c_str(), "r", stdin);
    freopen((s + ".out").c_str(), "w", stdout);
}

const int N = 2e5 + 2;
const int N1 = 1e6 + 2;
const int inf = 1e9 + 2;
const int INF = 1e18 + 2;
const int mod = 1e9 + 7;
int T = 1;

void solve()
{
    int n, k;
    string s;
    cin >>n >>k >>s;
    map<char, int> id;
    vector< int > g[4];
    id['J'] = 1;
    id['O'] = 2;
    id['I'] = 3;
    for( int i = 0; i < n; ++i )
        g[id[s[i]]].pb(i);
    g[1].pb(n + 1);
    g[2].pb(n + 1);
    g[3].pb(n + 1);
    int ans = inf;
    for( int i = 0; i + k < sz(g[1]); ++i )
    {
        int j = upper_bound(g[2].begin(), g[2].end(), g[1][i + k - 1]) - g[2].begin();
        if( j + k >= sz(g[2]) )
            continue;
        int l = upper_bound(g[3].begin(), g[3].end(), g[2][j + k - 1]) - g[3].begin();
        if( l + k >= sz(g[3]) )
            continue;
        ans = min(ans, g[3][l + k - 1] - g[1][i] + 1);
    }
    cout <<(ans == inf ? -1 : ans - k * 3);
}

signed main()
{
    file("");
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    // cin >>T;
    while( T-- )
        solve();
}

Compilation message (stderr)

ho_t2.cpp: In function 'void file(std::string)':
ho_t2.cpp:29:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   29 |     freopen((s + ".in").c_str(), "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
ho_t2.cpp:30:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |     freopen((s + ".out").c_str(), "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

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