#include <bits/stdc++.h>
using namespace std;
const int N = 8e3 + 5;
int Pref[2][N];
int Check(int n, int k)
{
int missed = Pref[0][n];
int taken = Pref[1][n];
int left = k - missed;
taken = max(0, taken - left);
if(left < 0)
return -1;
return n - taken;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
int n, k, t;
string A;
cin >> n >> k >> t;
cin >> A;
A = "#" + A;
for(int i = 1; i <= n; i++) {
Pref[0][i] = Pref[0][i-1] + (A[i] == '1');
Pref[1][i] = Pref[1][i-1] + (A[i] == '2');
}
int sol = Check(n, k);
for(int i = t+1; i <= n; i++) {
for(int j = i; j <= n-t; j++) {
int missed = Pref[0][i-1] + (Pref[0][n] - Pref[0][j]);
missed += (Pref[1][i-1] - Pref[1][i-t-1]) + (Pref[1][j+t] - Pref[1][j]);
int taken = Pref[1][i-t-1] + (Pref[1][n] - Pref[1][j+t]);
int left = k - missed;
taken = max(0, taken - left);
if(left < 0)
continue;
sol = max(sol, (n - (j+t) + (i-t-1)) - taken);
}
}
cout << sol << "\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; const int N = 8e3 + 5; int Pref[2][N]; int Check(int n, int k) { int missed = Pref[0][n]; int taken = Pref[1][n]; int left = k - missed; taken = max(0, taken - left); if(left < 0) return -1; return n - taken; } int main() { ios_base::sync_with_stdio(false); cin.tie(0); int n, k, t; string A; cin >> n >> k >> t; cin >> A; A = "#" + A; for(int i = 1; i <= n; i++) { Pref[0][i] = Pref[0][i-1] + (A[i] == '1'); Pref[1][i] = Pref[1][i-1] + (A[i] == '2'); } int sol = Check(n, k); for(int i = t+1; i <= n; i++) { for(int j = i; j <= n-t; j++) { int missed = Pref[0][i-1] + (Pref[0][n] - Pref[0][j]); missed += (Pref[1][i-1] - Pref[1][i-t-1]) + (Pref[1][j+t] - Pref[1][j]); int taken = Pref[1][i-t-1] + (Pref[1][n] - Pref[1][j+t]); int left = k - missed; taken = max(0, taken - left); if(left < 0) continue; sol = max(sol, (n - (j+t) + (i-t-1)) - taken); } } cout << sol << "\n"; return 0; } |
English