#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) {}
#endif
#define x first
#define y second
#define ir(a, x, b) ((a) <= (x) && (x) <= (b))
#define vec vector
#define rep(i, a, b) for (int i = a; i < (b); ++i)
#define all(x) (x).begin(), (x).end()
using ll = long long;
int main() {
cin.tie(0)->sync_with_stdio(0);
int N, K, t;
cin >> N >> K >> t;
vec<int> pw(N+1), pr(N+1);
string s; cin >> s;
rep (n, 0, N) {
pw[n+1] = pw[n];
pr[n+1] = pr[n];
if (s[n] == '1') pw[n+1]++;
if (s[n] == '2') pr[n+1]++;
}
int best = -1;
int min = max(0, pw[N] + pr[N] - K);
debug(min);
if (pr[N] >= min) {
best = max(best, N - min);
}
rep (j, 0, N+1) rep (i, 0, j) {
// at work [i, j)
// can drive there
if (i - t < 0) continue;
if (j + t > N) continue;
debug(i, j);
int atwork = pw[j] - pw[i] + pr[j] - pr[i];
int athome = pr[i-t] + (pr[N] - pr[j+t]);
debug(atwork, athome);
if (atwork + athome < min) continue;
int doathome = min - atwork;
best = max(best, N - (j - i + 2*t) - doathome);
}
cout << best << "\n";
return 0;
}
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 | #include <bits/stdc++.h> using namespace std; #ifdef LOCAL #include "debug.h" #else #define debug(...) {} #endif #define x first #define y second #define ir(a, x, b) ((a) <= (x) && (x) <= (b)) #define vec vector #define rep(i, a, b) for (int i = a; i < (b); ++i) #define all(x) (x).begin(), (x).end() using ll = long long; int main() { cin.tie(0)->sync_with_stdio(0); int N, K, t; cin >> N >> K >> t; vec<int> pw(N+1), pr(N+1); string s; cin >> s; rep (n, 0, N) { pw[n+1] = pw[n]; pr[n+1] = pr[n]; if (s[n] == '1') pw[n+1]++; if (s[n] == '2') pr[n+1]++; } int best = -1; int min = max(0, pw[N] + pr[N] - K); debug(min); if (pr[N] >= min) { best = max(best, N - min); } rep (j, 0, N+1) rep (i, 0, j) { // at work [i, j) // can drive there if (i - t < 0) continue; if (j + t > N) continue; debug(i, j); int atwork = pw[j] - pw[i] + pr[j] - pr[i]; int athome = pr[i-t] + (pr[N] - pr[j+t]); debug(atwork, athome); if (atwork + athome < min) continue; int doathome = min - atwork; best = max(best, N - (j - i + 2*t) - doathome); } cout << best << "\n"; return 0; } |
English