Submission #520634

#	Time	Username	Problem	Language	Result	Execution time	Memory
520634		aris12345678	JJOOII 2 (JOI20_ho_t2)	C++14	100 / 100	13 ms	2004 KiB

ho_t2

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

int main() {
    int n, k;
    scanf("%d %d", &n, &k);
    string s;
    cin >> s;
    vector<int> a, b, c;
    for(int i = 0; i < n; i++) {
        if(s[i] == 'J')
            a.push_back(i);
        else if(s[i] == 'O')
            b.push_back(i);
        else
            c.push_back(i);
    }
    int ans = INT_MAX;
    for(int i = k-1; i < (int)a.size(); i++) {
        // cout << a[i] << "\n";
        int j = upper_bound(b.begin(), b.end(), a[i])-b.begin();
        // cout << b[j] << "\n";
        if(j+k-1 >= (int)b.size()) break;
        int u = upper_bound(c.begin(), c.end(), b[j+k-1])-c.begin();
        if(u+k-1 >= (int)c.size()) break;
        // cout << a[i-k+1] << " " << c[u+k-1] << "\n";
        ans = min(ans, (c[u+k-1]-a[i-k+1]+1)-3*k);
    }
    if(ans == INT_MAX)
        printf("-1\n");
    else
        printf("%d\n", ans);
    return 0;
}

Compilation message (stderr)

ho_t2.cpp: In function 'int main()':
ho_t2.cpp:6:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
    6 |     scanf("%d %d", &n, &k);
      |     ~~~~~^~~~~~~~~~~~~~~~~

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