#include <bits/stdc++.h>
using ll = long long;
using sz = size_t;
using namespace std;
// make the code less c++-readable:
template<class T> using v = vector<T>;
template<class T> using vv = v<v<T>>;
using vi = v<int>; using vll = v<ll>; using vvi = vv<int>; using vvll = vv<ll>;
// hai loading utilities
#define $T template<class T>
#define $Ts template<class... T>
$T T Load() { T v; cin >> v; return v; }
$T auto Loads(int n) { v<T> v; v.reserve(n); while(n--) v.emplace_back(Load<T>()); return v; }
$T auto Loads() { return Loads<T>(Load<int>()); }
template<class T, int N> auto Loada() { array<T, N> a; for (T& v: a) v = Load<T>(); return a; }
$Ts auto Cols(int rows) { tuple<v<T>...> t; while(rows--) [&]<sz... I>(index_sequence<I...>){(std::get<I>(t).push_back(Load<T>()), ...);}(index_sequence_for<T...>{}); return t; }
//$Ts auto Rows(int rows) { v<tuple<T...>> v; while(rows--) { v.emplace_back(Load<T>()...); } return v; } bugged :(
struct _aIV { $T operator vector<T>() { return Loads<T>(n); } sz n; };
struct _aI { $T operator T() { return Load<T>(); } _aIV operator()(sz n) { return {n}; } }; static inline _aI $; /* int N = $; vi Y = $(N); */
#define MAKE_LOADER(T, alias) \
T alias() { return Load<T>(); } /* int x = Int(); */\
auto alias##s() { return Loads<T>(); } /* vector<> xs = Ints(); */\
auto alias##s(int n) { return Loads<T>(n); } /* vector<> xs = Ints(7); */\
template<int N> auto alias##a() { return Loada<T, N>(); } /* array<> xs = Inta<7>(); */\
// line intentionally left blank
MAKE_LOADER(int, Int)
MAKE_LOADER(long long, LL)
MAKE_LOADER(char, Char)
MAKE_LOADER(string, String)
// kthxbye
void test() {
int ans = -1;
const int N = $;
const int K = $;
const int T = $;
vector<int> pref_B(N+1); // biuro
vector<int> pref_Z(N+1); // zdalnie
const string plan = $;
for (int i = 0; i < plan.size(); ++i) {
pref_B[i+1] = pref_B[i] + int(plan[i] == '1');
pref_Z[i+1] = pref_Z[i] + int(plan[i] == '2');
}
// Wariant: nie jadę
if (pref_B[N] <= K) {
int opuszczonych = pref_B[N];
int jeszcze_można_opuścić = K - opuszczonych;
int trzeba_zdalnych = max(0, pref_Z[N] - jeszcze_można_opuścić);
ans = N - trzeba_zdalnych;
}
// Wariant: jadę
for (int wyjazd_do_biura = 0; wyjazd_do_biura + 2*T <= N; ++wyjazd_do_biura) {
for (int wyjazd_do_domu = wyjazd_do_biura+T; wyjazd_do_domu + T <= N; ++wyjazd_do_domu) {
int opuszczonych =
// po drodze (zdalnych):
+ (pref_Z[wyjazd_do_biura + T] - pref_Z[wyjazd_do_biura])
+ (pref_Z[wyjazd_do_domu + T] - pref_Z[wyjazd_do_domu])
// po drodze i w domu (biurnych)
+ (pref_B[wyjazd_do_biura + T] - pref_B[0])
+ (pref_B[N] - pref_B[wyjazd_do_domu])
;
if (opuszczonych > K) continue;
int jeszcze_można_opuścić = K - opuszczonych;
int zdalnych_w_domu = pref_Z[wyjazd_do_biura] + (pref_Z[N] - pref_Z[wyjazd_do_domu + T]);
zdalnych_w_domu = max(0, zdalnych_w_domu - jeszcze_można_opuścić);
int godzin_w_domu = wyjazd_do_biura + (N - (wyjazd_do_domu+T));
int subans = godzin_w_domu - zdalnych_w_domu;
ans = max(ans, subans);
}
}
cout << ans;
}
int main() {
test();
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 | #include <bits/stdc++.h> using ll = long long; using sz = size_t; using namespace std; // make the code less c++-readable: template<class T> using v = vector<T>; template<class T> using vv = v<v<T>>; using vi = v<int>; using vll = v<ll>; using vvi = vv<int>; using vvll = vv<ll>; // hai loading utilities #define $T template<class T> #define $Ts template<class... T> $T T Load() { T v; cin >> v; return v; } $T auto Loads(int n) { v<T> v; v.reserve(n); while(n--) v.emplace_back(Load<T>()); return v; } $T auto Loads() { return Loads<T>(Load<int>()); } template<class T, int N> auto Loada() { array<T, N> a; for (T& v: a) v = Load<T>(); return a; } $Ts auto Cols(int rows) { tuple<v<T>...> t; while(rows--) [&]<sz... I>(index_sequence<I...>){(std::get<I>(t).push_back(Load<T>()), ...);}(index_sequence_for<T...>{}); return t; } //$Ts auto Rows(int rows) { v<tuple<T...>> v; while(rows--) { v.emplace_back(Load<T>()...); } return v; } bugged :( struct _aIV { $T operator vector<T>() { return Loads<T>(n); } sz n; }; struct _aI { $T operator T() { return Load<T>(); } _aIV operator()(sz n) { return {n}; } }; static inline _aI $; /* int N = $; vi Y = $(N); */ #define MAKE_LOADER(T, alias) \ T alias() { return Load<T>(); } /* int x = Int(); */\ auto alias##s() { return Loads<T>(); } /* vector<> xs = Ints(); */\ auto alias##s(int n) { return Loads<T>(n); } /* vector<> xs = Ints(7); */\ template<int N> auto alias##a() { return Loada<T, N>(); } /* array<> xs = Inta<7>(); */\ // line intentionally left blank MAKE_LOADER(int, Int) MAKE_LOADER(long long, LL) MAKE_LOADER(char, Char) MAKE_LOADER(string, String) // kthxbye void test() { int ans = -1; const int N = $; const int K = $; const int T = $; vector<int> pref_B(N+1); // biuro vector<int> pref_Z(N+1); // zdalnie const string plan = $; for (int i = 0; i < plan.size(); ++i) { pref_B[i+1] = pref_B[i] + int(plan[i] == '1'); pref_Z[i+1] = pref_Z[i] + int(plan[i] == '2'); } // Wariant: nie jadę if (pref_B[N] <= K) { int opuszczonych = pref_B[N]; int jeszcze_można_opuścić = K - opuszczonych; int trzeba_zdalnych = max(0, pref_Z[N] - jeszcze_można_opuścić); ans = N - trzeba_zdalnych; } // Wariant: jadę for (int wyjazd_do_biura = 0; wyjazd_do_biura + 2*T <= N; ++wyjazd_do_biura) { for (int wyjazd_do_domu = wyjazd_do_biura+T; wyjazd_do_domu + T <= N; ++wyjazd_do_domu) { int opuszczonych = // po drodze (zdalnych): + (pref_Z[wyjazd_do_biura + T] - pref_Z[wyjazd_do_biura]) + (pref_Z[wyjazd_do_domu + T] - pref_Z[wyjazd_do_domu]) // po drodze i w domu (biurnych) + (pref_B[wyjazd_do_biura + T] - pref_B[0]) + (pref_B[N] - pref_B[wyjazd_do_domu]) ; if (opuszczonych > K) continue; int jeszcze_można_opuścić = K - opuszczonych; int zdalnych_w_domu = pref_Z[wyjazd_do_biura] + (pref_Z[N] - pref_Z[wyjazd_do_domu + T]); zdalnych_w_domu = max(0, zdalnych_w_domu - jeszcze_można_opuścić); int godzin_w_domu = wyjazd_do_biura + (N - (wyjazd_do_domu+T)); int subans = godzin_w_domu - zdalnych_w_domu; ans = max(ans, subans); } } cout << ans; } int main() { test(); return 0; } |
English