#include <bits/stdc++.h> #define st first #define nd second #define pb push_back #define ub upper_bound #define lb lower_bound #define fr(i,n) for(int i=0; i<(n); i++) #define frx(i,a,b) for(int i=(a); i<=(b); i++) #define ok(cond) cout << ((cond) ? "Yes" : "No") << "\n"; #define watch(x) cerr << (#x) << " == " << (x) << endl; using namespace std; typedef long long ll; typedef long double ld; typedef const int ci; typedef const ll cll; typedef const ld cld; typedef pair<int,int> pii; typedef set<int> si; typedef vector<int> vi; typedef vector<pair<int,int>> vpii; typedef vector<pair<int,ll>> vpill; template<typename T1, typename T2> istream& operator>>(istream& is, pair<T1,T2>& p) { return is >> p.st >> p.nd; } template<typename T1, typename T2> ostream& operator<<(ostream& os, pair<T1,T2>& p) { return os << p.st << " " << p.nd; } template<typename T> ostream& operator<<(ostream& os, vector<T>& v) { for (T x : v) { os << x << " "; } return os; } template<typename T> ostream& operator<<(ostream& os, set<T>& s) { for (T x : s) { os << x << " "; } return os; } const int N = 8e3; int cnt[3][N+5]; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, t; cin>>n>>k>>t; fr(i,n) { char c; cin>>c; cnt[c-'1'][i+1] = 1; fr(j,3) { cnt[j][i+1] += cnt[j][i]; } } int min_meets = cnt[0][n] + cnt[1][n] - k; if (cnt[0][n] <= k) { cout << cnt[2][n] + min(cnt[0][n] + cnt[1][n], k) << "\n"; return 0; } int mmax = -1; frx(beg, t+1, n-t) { frx(end, beg, n-t) { int in_office_meets = cnt[0][end] - cnt[0][beg-1] + cnt[1][end] - cnt[1][beg-1]; int remote_meets = cnt[1][n] - cnt[1][end+t] + cnt[1][beg-t-1]; if (in_office_meets + remote_meets >= min_meets) { remote_meets = max(0, min_meets - in_office_meets); mmax = max(mmax, n - (end+t) + (beg-t-1) - remote_meets); } } } cout << mmax << "\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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | #include <bits/stdc++.h> #define st first #define nd second #define pb push_back #define ub upper_bound #define lb lower_bound #define fr(i,n) for(int i=0; i<(n); i++) #define frx(i,a,b) for(int i=(a); i<=(b); i++) #define ok(cond) cout << ((cond) ? "Yes" : "No") << "\n"; #define watch(x) cerr << (#x) << " == " << (x) << endl; using namespace std; typedef long long ll; typedef long double ld; typedef const int ci; typedef const ll cll; typedef const ld cld; typedef pair<int,int> pii; typedef set<int> si; typedef vector<int> vi; typedef vector<pair<int,int>> vpii; typedef vector<pair<int,ll>> vpill; template<typename T1, typename T2> istream& operator>>(istream& is, pair<T1,T2>& p) { return is >> p.st >> p.nd; } template<typename T1, typename T2> ostream& operator<<(ostream& os, pair<T1,T2>& p) { return os << p.st << " " << p.nd; } template<typename T> ostream& operator<<(ostream& os, vector<T>& v) { for (T x : v) { os << x << " "; } return os; } template<typename T> ostream& operator<<(ostream& os, set<T>& s) { for (T x : s) { os << x << " "; } return os; } const int N = 8e3; int cnt[3][N+5]; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, t; cin>>n>>k>>t; fr(i,n) { char c; cin>>c; cnt[c-'1'][i+1] = 1; fr(j,3) { cnt[j][i+1] += cnt[j][i]; } } int min_meets = cnt[0][n] + cnt[1][n] - k; if (cnt[0][n] <= k) { cout << cnt[2][n] + min(cnt[0][n] + cnt[1][n], k) << "\n"; return 0; } int mmax = -1; frx(beg, t+1, n-t) { frx(end, beg, n-t) { int in_office_meets = cnt[0][end] - cnt[0][beg-1] + cnt[1][end] - cnt[1][beg-1]; int remote_meets = cnt[1][n] - cnt[1][end+t] + cnt[1][beg-t-1]; if (in_office_meets + remote_meets >= min_meets) { remote_meets = max(0, min_meets - in_office_meets); mmax = max(mmax, n - (end+t) + (beg-t-1) - remote_meets); } } } cout << mmax << "\n"; return 0; } |