Submission #733502

# Submission time Handle Problem Language Result Execution time Memory
733502 2023-05-01T00:52:31 Z Pring JJOOII 2 (JOI20_ho_t2) C++14
100 / 100
6 ms 3460 KB
#include <bits/stdc++.h>
using namespace std;

#define int long long
typedef pair<int, int> pii;

const int MXN = 200005;
int n, k, cid[256], ID[3][MXN], pid[3], jx[MXN], ix[MXN], ans = LLONG_MAX;
string s;

void init() {
    cid['J'] = 0;
    cid['O'] = 1;
    cid['I'] = 2;
}

void GETX() {
    ID[0][pid[0]] = LLONG_MAX;
    ID[2][pid[2]] = LLONG_MAX;
    int j = 0, i = 0;
    for (int o = 0; o < pid[1]; o++) {
        while (ID[0][j] < ID[1][o]) j++;
        jx[o] = j - 1;
        while (ID[2][i] < ID[1][o]) i++;
        ix[o] = i;
    }
}

int f(int from, int to, int space) {
    return to - from - space + 1;
}

int solve() {
    cin >> n >> k >> s;
    for (int i = 0; i < n; i++) {
        ID[cid[s[i]]][pid[cid[s[i]]]] = i;
        pid[cid[s[i]]]++;
    }
    for (int i = 0; i < 3; i++) if (pid[i] < k) return -1;
    // for (int i = 0; i < 3; i++) {
    //     for (int j = 0; j < pid[i]; j++) {
    //         cout << ID[i][j] << ' ';
    //     }
    //     cout << endl;
    // }
    GETX();
    // for (int i = 0; i < pid[1]; i++) {
    //     cout << jx[i] << ' ' << ix[i] << endl;
    // }
    for (int i = 0; i <= pid[1] - k; i++) {
        if (jx[i] - k + 1 < 0) continue;
        if (ix[i + k - 1] + k - 1 >= pid[2]) continue;
        // cout << i << " PRING" << endl;
        ans = min(ans, f(ID[1][i], ID[1][i + k - 1], k) + f(ID[0][jx[i] - k + 1], ID[1][i], k + 1) + f(ID[1][i + k - 1], ID[2][ix[i + k - 1] + k - 1], k + 1));
    }
    if (ans == LLONG_MAX) return -1;
    return ans;
}

int32_t main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    init();
    cout << solve() << endl;
    return 0;
}

Compilation message

ho_t2.cpp: In function 'long long int solve()':
ho_t2.cpp:36:20: warning: array subscript has type 'char' [-Wchar-subscripts]
   36 |         ID[cid[s[i]]][pid[cid[s[i]]]] = i;
      |                    ^
ho_t2.cpp:36:35: warning: array subscript has type 'char' [-Wchar-subscripts]
   36 |         ID[cid[s[i]]][pid[cid[s[i]]]] = i;
      |                                   ^
ho_t2.cpp:37:21: warning: array subscript has type 'char' [-Wchar-subscripts]
   37 |         pid[cid[s[i]]]++;
      |                     ^
# Verdict Execution time Memory Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 0 ms 340 KB Output is correct
5 Correct 0 ms 340 KB Output is correct
6 Correct 0 ms 340 KB Output is correct
7 Correct 0 ms 340 KB Output is correct
8 Correct 1 ms 328 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 1 ms 340 KB Output is correct
11 Correct 1 ms 332 KB Output is correct
12 Correct 1 ms 340 KB Output is correct
13 Correct 1 ms 336 KB Output is correct
14 Correct 1 ms 340 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 0 ms 340 KB Output is correct
5 Correct 0 ms 340 KB Output is correct
6 Correct 0 ms 340 KB Output is correct
7 Correct 0 ms 340 KB Output is correct
8 Correct 1 ms 328 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 1 ms 340 KB Output is correct
11 Correct 1 ms 332 KB Output is correct
12 Correct 1 ms 340 KB Output is correct
13 Correct 1 ms 336 KB Output is correct
14 Correct 1 ms 340 KB Output is correct
15 Correct 1 ms 340 KB Output is correct
16 Correct 1 ms 340 KB Output is correct
17 Correct 1 ms 328 KB Output is correct
18 Correct 1 ms 340 KB Output is correct
19 Correct 1 ms 332 KB Output is correct
20 Correct 1 ms 340 KB Output is correct
21 Correct 1 ms 332 KB Output is correct
22 Correct 1 ms 340 KB Output is correct
23 Correct 1 ms 336 KB Output is correct
24 Correct 1 ms 340 KB Output is correct
25 Correct 1 ms 332 KB Output is correct
26 Correct 1 ms 340 KB Output is correct
27 Correct 1 ms 340 KB Output is correct
28 Correct 0 ms 340 KB Output is correct
29 Correct 1 ms 340 KB Output is correct
30 Correct 1 ms 340 KB Output is correct
31 Correct 1 ms 340 KB Output is correct
32 Correct 1 ms 340 KB Output is correct
33 Correct 1 ms 340 KB Output is correct
34 Correct 1 ms 328 KB Output is correct
35 Correct 1 ms 340 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 0 ms 340 KB Output is correct
5 Correct 0 ms 340 KB Output is correct
6 Correct 0 ms 340 KB Output is correct
7 Correct 0 ms 340 KB Output is correct
8 Correct 1 ms 328 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 1 ms 340 KB Output is correct
11 Correct 1 ms 332 KB Output is correct
12 Correct 1 ms 340 KB Output is correct
13 Correct 1 ms 336 KB Output is correct
14 Correct 1 ms 340 KB Output is correct
15 Correct 1 ms 340 KB Output is correct
16 Correct 1 ms 340 KB Output is correct
17 Correct 1 ms 328 KB Output is correct
18 Correct 1 ms 340 KB Output is correct
19 Correct 1 ms 332 KB Output is correct
20 Correct 1 ms 340 KB Output is correct
21 Correct 1 ms 332 KB Output is correct
22 Correct 1 ms 340 KB Output is correct
23 Correct 1 ms 336 KB Output is correct
24 Correct 1 ms 340 KB Output is correct
25 Correct 1 ms 332 KB Output is correct
26 Correct 1 ms 340 KB Output is correct
27 Correct 1 ms 340 KB Output is correct
28 Correct 0 ms 340 KB Output is correct
29 Correct 1 ms 340 KB Output is correct
30 Correct 1 ms 340 KB Output is correct
31 Correct 1 ms 340 KB Output is correct
32 Correct 1 ms 340 KB Output is correct
33 Correct 1 ms 340 KB Output is correct
34 Correct 1 ms 328 KB Output is correct
35 Correct 1 ms 340 KB Output is correct
36 Correct 5 ms 3172 KB Output is correct
37 Correct 6 ms 3424 KB Output is correct
38 Correct 6 ms 3428 KB Output is correct
39 Correct 6 ms 3424 KB Output is correct
40 Correct 6 ms 3424 KB Output is correct
41 Correct 5 ms 3424 KB Output is correct
42 Correct 6 ms 3424 KB Output is correct
43 Correct 3 ms 2260 KB Output is correct
44 Correct 5 ms 2784 KB Output is correct
45 Correct 6 ms 3424 KB Output is correct
46 Correct 5 ms 3424 KB Output is correct
47 Correct 6 ms 3376 KB Output is correct
48 Correct 6 ms 3448 KB Output is correct
49 Correct 4 ms 2404 KB Output is correct
50 Correct 5 ms 3460 KB Output is correct
51 Correct 6 ms 3424 KB Output is correct
52 Correct 5 ms 3316 KB Output is correct
53 Correct 6 ms 3404 KB Output is correct
54 Correct 5 ms 3404 KB Output is correct
55 Correct 5 ms 3428 KB Output is correct
56 Correct 5 ms 3408 KB Output is correct