// #pragma GCC optimize("O3") // #pragma GCC optimize("Ofast") // #pragma GCC optimize("unroll-loops") // #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,bmi,bmi2,lzcnt,abm,mmx,avx,avx2,tune=native") #include <bits/stdc++.h> using namespace std; // #include <ext/pb_ds/assoc_container.hpp> // #include <ext/pb_ds/tree_policy.hpp> // using namespace __gnu_pbds; // template <typename T> // using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>; typedef long long ll; typedef unsigned long long ull; typedef long double ld; #define rep(i, j) for (int i = 0; i < ((int)(j)); i++) #define pb push_back #define pf push_front #define st first #define nd second #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() #define get_unique(x) x.erase(unique(all(x)), x.end()); const int N = 10'000; int pref1[N], pref2[N], pref3[N]; void TEST([[maybe_unused]] int testcase_i) { int n, k, t; cin >> n >> k >> t; string s; cin >> s; rep(i, n) { if (s[i] == '1') { pref1[i + 1] = 1; } if (s[i] == '2') { pref2[i + 1] = 1; } if (s[i] == '3') { pref3[i + 1] = 1; } pref1[i + 1] += pref1[i]; pref2[i + 1] += pref2[i]; pref3[i + 1] += pref3[i]; } int ans = -1; if (pref1[n] <= k) ans = pref3[n] + pref1[n] + min(pref2[n], k - pref1[n]); for (int l = 0; l < n; l++) { for (int r = l + t; r + t - 1 < n; r++) { int miss_l = pref1[l + t] - pref1[l] + pref2[l + t] - pref2[l]; int miss_r = pref1[r + t] - pref1[r] + pref2[r + t] - pref2[r]; int miss_oth = pref1[l] + pref1[n] - pref1[r + t]; int miss_tot = miss_l + miss_r + miss_oth; if (miss_tot > k) continue; int can_miss = pref2[l] + pref2[n] - pref2[r + t]; int free_time = pref3[l] + pref3[n] - pref3[r + t]; ans = max(ans, free_time + miss_oth + min(can_miss, k - miss_tot)); } } cout << ans; } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr); // cout << setprecision(20) << fixed; // srand(time(NULL)); // freopen("sorting.in", "r", stdin); // freopen("sorting.out", "w", stdout); int t = 1; // cin >> t; rep(i, t) { TEST(i); } 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 95 96 97 98 | // #pragma GCC optimize("O3") // #pragma GCC optimize("Ofast") // #pragma GCC optimize("unroll-loops") // #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,bmi,bmi2,lzcnt,abm,mmx,avx,avx2,tune=native") #include <bits/stdc++.h> using namespace std; // #include <ext/pb_ds/assoc_container.hpp> // #include <ext/pb_ds/tree_policy.hpp> // using namespace __gnu_pbds; // template <typename T> // using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>; typedef long long ll; typedef unsigned long long ull; typedef long double ld; #define rep(i, j) for (int i = 0; i < ((int)(j)); i++) #define pb push_back #define pf push_front #define st first #define nd second #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() #define get_unique(x) x.erase(unique(all(x)), x.end()); const int N = 10'000; int pref1[N], pref2[N], pref3[N]; void TEST([[maybe_unused]] int testcase_i) { int n, k, t; cin >> n >> k >> t; string s; cin >> s; rep(i, n) { if (s[i] == '1') { pref1[i + 1] = 1; } if (s[i] == '2') { pref2[i + 1] = 1; } if (s[i] == '3') { pref3[i + 1] = 1; } pref1[i + 1] += pref1[i]; pref2[i + 1] += pref2[i]; pref3[i + 1] += pref3[i]; } int ans = -1; if (pref1[n] <= k) ans = pref3[n] + pref1[n] + min(pref2[n], k - pref1[n]); for (int l = 0; l < n; l++) { for (int r = l + t; r + t - 1 < n; r++) { int miss_l = pref1[l + t] - pref1[l] + pref2[l + t] - pref2[l]; int miss_r = pref1[r + t] - pref1[r] + pref2[r + t] - pref2[r]; int miss_oth = pref1[l] + pref1[n] - pref1[r + t]; int miss_tot = miss_l + miss_r + miss_oth; if (miss_tot > k) continue; int can_miss = pref2[l] + pref2[n] - pref2[r + t]; int free_time = pref3[l] + pref3[n] - pref3[r + t]; ans = max(ans, free_time + miss_oth + min(can_miss, k - miss_tot)); } } cout << ans; } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr); // cout << setprecision(20) << fixed; // srand(time(NULL)); // freopen("sorting.in", "r", stdin); // freopen("sorting.out", "w", stdout); int t = 1; // cin >> t; rep(i, t) { TEST(i); } return 0; } |