// PA 2024 1B - Praca #include <iostream> #include <string> #include <vector> #include <algorithm> using namespace std; int main() { ios_base::sync_with_stdio(0); int n, k, t; int best, candidate, homeOffice, meetingsLive, meetingsOnline; char c; vector<char> plan; vector<int> morningMissed; vector<int> morningMissedMaybe; vector<int> eveningMissed; vector<int> eveningMissedMaybe; cin >> n; cin >> k; cin >> t; meetingsLive = 0; meetingsOnline = 0; for (int i = 0; i < n; ++i) { cin >> c; if (c == '1') { ++meetingsLive; } if (c == '2') { ++meetingsOnline; } plan.push_back(c); morningMissed.push_back(0); morningMissedMaybe.push_back(0); eveningMissed.push_back(0); eveningMissedMaybe.push_back(0); } //special case if (meetingsLive <= k) { if (k > meetingsLive + meetingsOnline) { homeOffice = 0; } else { homeOffice = meetingsLive + meetingsOnline - k; } cout << n - homeOffice << endl; return 0; } for (int i = 0; i < t; ++i) { if (plan[i] == '1' || plan[i] == '2') { ++morningMissed[0]; } if (plan[n - 1 - i] == '1' || plan[n - 1 - i] == '2') { ++eveningMissed[n - 1]; } } for (int i = 1; i < n - t + 1; ++i) { morningMissed[i] = morningMissed[i - 1]; morningMissedMaybe[i] = morningMissedMaybe[i - 1]; if (plan[i - 1] == '2') { --morningMissed[i]; ++morningMissedMaybe[i]; } if (plan[i + t - 1] == '1' || plan[i + t - 1] == '2') { ++morningMissed[i]; } } for (int i = n - 2; i >= t - 1; --i) { eveningMissed[i] = eveningMissed[i + 1]; eveningMissedMaybe[i] = eveningMissedMaybe[i + 1]; if (plan[i + 1] == '2') { --eveningMissed[i]; ++eveningMissedMaybe[i]; } if (plan[i - t + 1] == '1' || plan[i - t + 1] == '2') { ++eveningMissed[i]; } } best = -1; for (int i = 0; i < n - t + 1; ++i) { for (int j = n - 1; j >= t - 1; --j) { if (i + t - 1 >= j - t + 1) { break; } if (morningMissed[i] + eveningMissed[j] <= k) { homeOffice = (morningMissedMaybe[i] + eveningMissedMaybe[j]) - (k - (morningMissed[i] + eveningMissed[j])); if (homeOffice < 0) { homeOffice = 0; } candidate = n - (j - i + 1) - homeOffice; if (candidate > best) { best = candidate; } } } } cout << best << endl; 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 99 100 101 102 103 104 105 106 107 108 | // PA 2024 1B - Praca #include <iostream> #include <string> #include <vector> #include <algorithm> using namespace std; int main() { ios_base::sync_with_stdio(0); int n, k, t; int best, candidate, homeOffice, meetingsLive, meetingsOnline; char c; vector<char> plan; vector<int> morningMissed; vector<int> morningMissedMaybe; vector<int> eveningMissed; vector<int> eveningMissedMaybe; cin >> n; cin >> k; cin >> t; meetingsLive = 0; meetingsOnline = 0; for (int i = 0; i < n; ++i) { cin >> c; if (c == '1') { ++meetingsLive; } if (c == '2') { ++meetingsOnline; } plan.push_back(c); morningMissed.push_back(0); morningMissedMaybe.push_back(0); eveningMissed.push_back(0); eveningMissedMaybe.push_back(0); } //special case if (meetingsLive <= k) { if (k > meetingsLive + meetingsOnline) { homeOffice = 0; } else { homeOffice = meetingsLive + meetingsOnline - k; } cout << n - homeOffice << endl; return 0; } for (int i = 0; i < t; ++i) { if (plan[i] == '1' || plan[i] == '2') { ++morningMissed[0]; } if (plan[n - 1 - i] == '1' || plan[n - 1 - i] == '2') { ++eveningMissed[n - 1]; } } for (int i = 1; i < n - t + 1; ++i) { morningMissed[i] = morningMissed[i - 1]; morningMissedMaybe[i] = morningMissedMaybe[i - 1]; if (plan[i - 1] == '2') { --morningMissed[i]; ++morningMissedMaybe[i]; } if (plan[i + t - 1] == '1' || plan[i + t - 1] == '2') { ++morningMissed[i]; } } for (int i = n - 2; i >= t - 1; --i) { eveningMissed[i] = eveningMissed[i + 1]; eveningMissedMaybe[i] = eveningMissedMaybe[i + 1]; if (plan[i + 1] == '2') { --eveningMissed[i]; ++eveningMissedMaybe[i]; } if (plan[i - t + 1] == '1' || plan[i - t + 1] == '2') { ++eveningMissed[i]; } } best = -1; for (int i = 0; i < n - t + 1; ++i) { for (int j = n - 1; j >= t - 1; --j) { if (i + t - 1 >= j - t + 1) { break; } if (morningMissed[i] + eveningMissed[j] <= k) { homeOffice = (morningMissedMaybe[i] + eveningMissedMaybe[j]) - (k - (morningMissed[i] + eveningMissed[j])); if (homeOffice < 0) { homeOffice = 0; } candidate = n - (j - i + 1) - homeOffice; if (candidate > best) { best = candidate; } } } } cout << best << endl; return 0; } |