#include <bits/stdc++.h> #define ndl '\n' #define ll long long #define st first #define nd second #define debug(x) cout << #x << ": " << x << ndl #define all(x) (x).begin(), (x).end() using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); int n, k, t; cin>>n>>k>>t; vector<int> tab(n); char c; for(int i=0;i<n;i++) { cin>>c; tab[i] = c-'1'+1; } vector<int> w(n+1), z(n+1), b(n+1); for(int i=1;i<=n;i++) { w[i] = w[i-1] + (tab[i-1]<3?1:0); z[i] = z[i-1] + (tab[i-1]==2?1:0); b[i] = b[i-1] + (tab[i-1]==1?1:0); } // for(int i=0;i<=n;i++) { // cout<<w[i]<<" "; // } // cout<<'\n'; // for(int i=0;i<=n;i++) { // cout<<z[i]<<" "; // } if(w[n] - z[n] <= k) { cout<<n-w[n]+min(k, w[n])<<'\n'; return 0; } int ans = -1; for(int i=t+1;i<=n;i++) { for(int j=i;j+t<=n;j++) { int pominiete = w[i-1] - w[i-1-t] + w[min(n, j+t)] - w[j] // dojazdy + b[i-1-t] + b[n] - b[min(n, j+t)]; int do_pominiecia = k - pominiete; if(do_pominiecia < 0) { continue; } int zdalne = z[i-1-t] + z[n] - z[min(n, j+t)]; // if(n - w[n] + min(do_pominiecia, zdalne) > ans) { // cout<<i<<" "<<j<<'\n'; // cout<<pominiete<<'\n'; // cout<<do_pominiecia<<'\n'; // } ans = max(ans, n - (j - i + 1 + 2 * t) - zdalne + min(do_pominiecia, zdalne)); } } cout<<ans<<'\n'; // int juz_przegapione = 0; // int wolny = 0; // int zdalny = 0; // for(int i=1;i<=n;i++) { // int aktu_przegapione = juz_przegapione + w[i+t-1] - w[i-1]; // if(aktu_przegapione > k) { // break; // } // ans = max(ans, wolny + min(k-aktu_przegapione, zdalny)); // } }
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 74 75 | #include <bits/stdc++.h> #define ndl '\n' #define ll long long #define st first #define nd second #define debug(x) cout << #x << ": " << x << ndl #define all(x) (x).begin(), (x).end() using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); int n, k, t; cin>>n>>k>>t; vector<int> tab(n); char c; for(int i=0;i<n;i++) { cin>>c; tab[i] = c-'1'+1; } vector<int> w(n+1), z(n+1), b(n+1); for(int i=1;i<=n;i++) { w[i] = w[i-1] + (tab[i-1]<3?1:0); z[i] = z[i-1] + (tab[i-1]==2?1:0); b[i] = b[i-1] + (tab[i-1]==1?1:0); } // for(int i=0;i<=n;i++) { // cout<<w[i]<<" "; // } // cout<<'\n'; // for(int i=0;i<=n;i++) { // cout<<z[i]<<" "; // } if(w[n] - z[n] <= k) { cout<<n-w[n]+min(k, w[n])<<'\n'; return 0; } int ans = -1; for(int i=t+1;i<=n;i++) { for(int j=i;j+t<=n;j++) { int pominiete = w[i-1] - w[i-1-t] + w[min(n, j+t)] - w[j] // dojazdy + b[i-1-t] + b[n] - b[min(n, j+t)]; int do_pominiecia = k - pominiete; if(do_pominiecia < 0) { continue; } int zdalne = z[i-1-t] + z[n] - z[min(n, j+t)]; // if(n - w[n] + min(do_pominiecia, zdalne) > ans) { // cout<<i<<" "<<j<<'\n'; // cout<<pominiete<<'\n'; // cout<<do_pominiecia<<'\n'; // } ans = max(ans, n - (j - i + 1 + 2 * t) - zdalne + min(do_pominiecia, zdalne)); } } cout<<ans<<'\n'; // int juz_przegapione = 0; // int wolny = 0; // int zdalny = 0; // for(int i=1;i<=n;i++) { // int aktu_przegapione = juz_przegapione + w[i+t-1] - w[i-1]; // if(aktu_przegapione > k) { // break; // } // ans = max(ans, wolny + min(k-aktu_przegapione, zdalny)); // } } |