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
64
65
66
67
68
69
70
71
72
73
#include <bits/stdc++.h>
using namespace std;
int n, x, ile, wyn, essa, k, t, essa2, essa3, zapas;
int sum1[8004], sum2[8004], sum3[8004];
string s;

int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin >> n >> k >> t;
    cin >> s;
    for(int i = 1; i <= n; i++)
    {
        sum1[i] = sum1[i-1]; sum2[i] = sum2[i-1]; sum3[i] = sum3[i-1];
        if(s[i-1] == '1') // biuro
        {
            sum1[i]++;
        }
        else if(s[i-1] == '2') // spotkanie zdalne
        {
            sum2[i]++;
        }
        else
        {
            sum3[i]++;
        }
    }

    if(k >= sum2[n] + sum1[n])
    {
        cout << sum3[n] + sum2[n] + sum1[n]; return 0; // wtedy mozna w ogole nie byc w pracy
    }
    else if(k >= sum1[n])
    {
        wyn = n - (sum1[n]+sum2[n] -k); cout << wyn; return 0;
    }

    wyn = -1;
    for(int i = 1; i <= n; i++)
    {
        for(int y = 1; y <= i; y++) // poczatek przedzialu w ktorym juz jest w pracy
        {
            if(y - t < 1){continue;} // jesli 0
            if(i + t > n){continue;}
            ile = sum1[i] - sum1[y-1] + sum2[i] - sum2[y-1]; // tu w biurze
            ile = ile + sum2[y - t -1] + (sum2[n] - sum2[i+t]); // tu pracuje zdalnie
            if(ile + k < sum1[n] + sum2[n])
            {
                continue;
            }
            else
            {
                zapas = ile + k -sum1[n] - sum2[n]; // tyle jeszcze moge przeznaczyc
                essa = sum3[y - t -1] + (sum3[n] - sum3[i+t]);
                essa2 = sum2[y - t -1] + (sum2[n] - sum2[i+t]);
                essa3 =  sum1[y - t -1] + (sum1[n] - sum1[i+t]); // tyle ma spotkan w tym czasie wszystko na grind
                if((ile - essa2) + k < sum1[n]+sum2[n])
                {
                    essa2 = zapas; // tyle moge wykorzystac
                }
                essa = essa2 + essa3 + essa;
                wyn = max(essa, wyn);
            }
        }
    } // nie trzeba tez jechac kiedy caly czas pracuje sie zdalnie





        cout << wyn;

}