#include "bits/stdc++.h" using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, t; cin >> n >> k >> t; string s; cin >> s; s = '#' + s + '#'; vector <array <int, 4>> pref(s.size(), {0, 0, 0}), suf(s.size(), {0, 0, 0}); for (int i = 1; i < s.size(); i++) { pref[i] = pref[i - 1]; if (s[i] == '1') pref[i][1]++; if (s[i] == '2') pref[i][2]++; if (s[i] == '3') pref[i][3]++; } for (int i = s.size() - 2; i >= 0; i--) { suf[i] = suf[i + 1]; if (s[i] == '1') suf[i][1]++; if (s[i] == '2') suf[i][2]++; if (s[i] == '3') suf[i][3]++; } int ans = -1; if (pref[n][1] <= k) { int _k = k - pref[n][1]; ans = pref[n][3] + pref[n][1] + min(_k, pref[n][2]); } for (int i = 0; i <= n; i++) { for (int j = n + 1; j >= 0; j--) { if (j - i - 1 >= 2 * t) { int _k = k - (pref[i + t][2] - pref[i][2]) - (pref[i + t][1] - pref[i][1]); _k = _k - (pref[j - 1][2] - pref[j - 1 - t][2]) - (pref[j - 1][1] - pref[j - 1 - t][1]); _k -= pref[i][1]; _k -= suf[j][1]; if (_k < 0) continue; ans = max(ans, pref[i][3] + suf[j][3] + pref[i][1] + suf[j][1] + min(_k, pref[i][2] + suf[j][2])); } } } cout << ans << '\n'; }
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 | #include "bits/stdc++.h" using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, t; cin >> n >> k >> t; string s; cin >> s; s = '#' + s + '#'; vector <array <int, 4>> pref(s.size(), {0, 0, 0}), suf(s.size(), {0, 0, 0}); for (int i = 1; i < s.size(); i++) { pref[i] = pref[i - 1]; if (s[i] == '1') pref[i][1]++; if (s[i] == '2') pref[i][2]++; if (s[i] == '3') pref[i][3]++; } for (int i = s.size() - 2; i >= 0; i--) { suf[i] = suf[i + 1]; if (s[i] == '1') suf[i][1]++; if (s[i] == '2') suf[i][2]++; if (s[i] == '3') suf[i][3]++; } int ans = -1; if (pref[n][1] <= k) { int _k = k - pref[n][1]; ans = pref[n][3] + pref[n][1] + min(_k, pref[n][2]); } for (int i = 0; i <= n; i++) { for (int j = n + 1; j >= 0; j--) { if (j - i - 1 >= 2 * t) { int _k = k - (pref[i + t][2] - pref[i][2]) - (pref[i + t][1] - pref[i][1]); _k = _k - (pref[j - 1][2] - pref[j - 1 - t][2]) - (pref[j - 1][1] - pref[j - 1 - t][1]); _k -= pref[i][1]; _k -= suf[j][1]; if (_k < 0) continue; ans = max(ans, pref[i][3] + suf[j][3] + pref[i][1] + suf[j][1] + min(_k, pref[i][2] + suf[j][2])); } } } cout << ans << '\n'; } |