Submission #245415

#	Time	Username	Problem	Language	Result	Execution time	Memory
245415		kevlee	JJOOII 2 (JOI20_ho_t2)	C++17	100 / 100	14 ms	1700 KiB

ho_t2

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mod 1000000007
#define h1 7897897897897897
#define h2 7897466719774591
#define b1 98762051
#define b2 98765431
#define inf 1000000000
#define pi 3.1415926535897932384626
#define LMAX 9223372036854775807
#define ll long long
#define fi first
#define se second
#define pii pair<int, int>
#define pll pair<ll, ll>
#define vi vector<int>
#define vl vector<ll>
#define vp vector<pii>
#define SET(a, b) memset(a, b, sizeof(a));
#define all(x) (x).begin(), (x).end()
#define flush fflush(stdout)
#define debug printf("Hello\n")
#define FOR(i, a, b) for (int i = (a); i <= (b); i++)
#define FORD(i, a, b) for (int i = (a); i >= (b); i--)
//priority_queue <int, vector <int>, greater <int> > pq;
int n, m, k, x, y, sum, cnt, ans = inf, a[200005], b[200005], c[200005];
string s;
int jnn, on, in;
bool found;
int main () {
  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cin >> n >> k >> s;
  FOR(i, 0, n-1) {
    if (s[i] == 'J') {
      a[jnn++] = i;
    } else if (s[i] == 'O') {
      b[on++] = i;
    } else {
      c[in++] = i;
    }
  } 
  FOR(i, 0, jnn-1) {
    if (i + k - 1 >= jnn) break;
    int pos = upper_bound(b, b+on, a[i+k-1]) - b;
    if (pos == on) break;
    if (pos + k - 1 >= on) break;
    pos = upper_bound(c, c+in, b[pos+k-1]) - c;
    if (pos == in) break;
    if (pos + k - 1 >= in) break;
    pos = c[pos+k-1];
    found = true;
    ans = min(ans, pos - a[i] + 1 - 3 * k);
  }
  if (!found) cout << "-1" << endl;
  else cout << ans << endl;
  return 0;
}

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