#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; } |