1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#include <bits/stdc++.h>
using namespace std;

int wbiurze[8007];
int zdalne[8007];

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    
    int n, t, k;
    cin >> n >> k >> t;

    for (int i = 1; i <= n; i++) {
        char c;
        cin >> c;

        wbiurze[i] = wbiurze[i - 1];
        zdalne[i] = zdalne[i - 1];
        if (c == '1') {
            wbiurze[i]++;
        }
        else if (c == '2') {
            zdalne[i]++;
        }
    }

    // for (int i = 1; i <= n; i++) {
    //     cout << wbiurze[i] << " ";
    // }
    // cout << "\n";
    // for (int i = 1; i <= n; i++) {
    //     cout << zdalne[i] << " ";
    // }
    // cout << "\n";
    // cout << "k = " << k << "\n";

    int wynik = -1;
    if (wbiurze[n] <= k) {
        // cout << "dupa";
        wynik = max(wynik, n - zdalne[n] + min(k - wbiurze[n], zdalne[n]));
    }

    // cout << "wynik = " << wynik << "\n";

    for (int i = 0; i <= n - t; i++) {
        for (int j = i + t; j <= n - t; j++) {
            int koszt = wbiurze[i + t] // w biurze przed wyjściem i w trakcie podróży
                + zdalne[i + t] - zdalne[i] // zdalne w trakcie podróży
                + wbiurze[n] - wbiurze[j] // w biurze po powrocie i w trakcie podróży
                + zdalne[j + t] - zdalne[j]; // zdalne w trakcie podróży

            int ile = k - koszt; // tyle może jeszcze opuścić
            if (ile >= 0) {
                int wolne = i - zdalne[i] + (n - (j + t)) - (zdalne[n] - zdalne[j + t]) + min(ile, zdalne[i] + zdalne[n] - zdalne[j + t]);
                wynik = max(wynik, wolne);
            }
        }
    }

    cout << wynik << "\n";

}