#include <bits/stdc++.h>
typedef bool bool_t;
typedef char char_t;
typedef long long int_t;
typedef std::string str_t;
typedef std::vector<bool_t> vecbool_t;
typedef std::vector<int_t> vecint_t;
typedef std::vector<vecint_t> vecvecint_t;
typedef std::vector<vecvecint_t> vecvecvecint_t;
int_t n, k, t;
str_t s;
void s2();
int32_t main()
{
std::ios_base::sync_with_stdio(0);
std::cin.tie(0);
std::cin >> n >> k >> t >> s;
s2();
return 0;
}
void s2()
{
int_t answer;
vecint_t meetings_on_prefix(n + 1, 0);
vecvecvecint_t dp(n + 1, vecvecint_t(k + 1, vecint_t(2, LLONG_MIN)));
for(int_t i = 0; i < n; i++)
meetings_on_prefix[i + 1] = meetings_on_prefix[i] + (s[i] == '1' || s[i] == '2');
for(int_t i = 0; i <= k; i++)
dp[0][i][0] = 0;
for(int_t i = 1; i <= n; i++)
{
for(int_t j = 0; j <= k; j++)
{
int_t p = 0;
if((i - t) >= 0)
p = meetings_on_prefix[i] - meetings_on_prefix[i - t];
if(s[i - 1] == '1')
{
if(j - 1 >= 0)
dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j - 1][0] + 1);
if(i - t >= 0 && j - p >= 0)
dp[i][j][0] = std::max(dp[i][j][0], dp[i - t][j - p][1]);
dp[i][j][1] = std::max(dp[i][j][1], dp[i - 1][j][1]);
if(i - t >= 0 && j - p >= 0)
dp[i][j][1] = std::max(dp[i][j][1], dp[i - t][j - p][0]);
}
if(s[i - 1] == '2')
{
dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j][0]);
if(j - 1 >= 0)
dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j - 1][0] + 1);
if(i - t >= 0 && j - p >= 0)
dp[i][j][0] = std::max(dp[i][j][0], dp[i - t][j - p][1]);
dp[i][j][1] = std::max(dp[i][j][1], dp[i - 1][j][1]);
if(i - t >= 0 && j - p >= 0)
dp[i][j][1] = std::max(dp[i][j][1], dp[i - t][j - p][0]);
}
if(s[i - 1] == '3')
{
dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j][0] + 1);
if(i - t >= 0 && j - p >= 0)
dp[i][j][0] = std::max(dp[i][j][0], dp[i - t][j - p][1]);
dp[i][j][1] = std::max(dp[i][j][1], dp[i - 1][j][1]);
if(i - t >= 0 && j - p >= 0)
dp[i][j][1] = std::max(dp[i][j][1], dp[i - t][j - p][0]);
}
}
}
answer = dp[n][k][0];
if(answer < 0)
answer = -1;
std::cout << answer << '\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 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 | #include <bits/stdc++.h> typedef bool bool_t; typedef char char_t; typedef long long int_t; typedef std::string str_t; typedef std::vector<bool_t> vecbool_t; typedef std::vector<int_t> vecint_t; typedef std::vector<vecint_t> vecvecint_t; typedef std::vector<vecvecint_t> vecvecvecint_t; int_t n, k, t; str_t s; void s2(); int32_t main() { std::ios_base::sync_with_stdio(0); std::cin.tie(0); std::cin >> n >> k >> t >> s; s2(); return 0; } void s2() { int_t answer; vecint_t meetings_on_prefix(n + 1, 0); vecvecvecint_t dp(n + 1, vecvecint_t(k + 1, vecint_t(2, LLONG_MIN))); for(int_t i = 0; i < n; i++) meetings_on_prefix[i + 1] = meetings_on_prefix[i] + (s[i] == '1' || s[i] == '2'); for(int_t i = 0; i <= k; i++) dp[0][i][0] = 0; for(int_t i = 1; i <= n; i++) { for(int_t j = 0; j <= k; j++) { int_t p = 0; if((i - t) >= 0) p = meetings_on_prefix[i] - meetings_on_prefix[i - t]; if(s[i - 1] == '1') { if(j - 1 >= 0) dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j - 1][0] + 1); if(i - t >= 0 && j - p >= 0) dp[i][j][0] = std::max(dp[i][j][0], dp[i - t][j - p][1]); dp[i][j][1] = std::max(dp[i][j][1], dp[i - 1][j][1]); if(i - t >= 0 && j - p >= 0) dp[i][j][1] = std::max(dp[i][j][1], dp[i - t][j - p][0]); } if(s[i - 1] == '2') { dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j][0]); if(j - 1 >= 0) dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j - 1][0] + 1); if(i - t >= 0 && j - p >= 0) dp[i][j][0] = std::max(dp[i][j][0], dp[i - t][j - p][1]); dp[i][j][1] = std::max(dp[i][j][1], dp[i - 1][j][1]); if(i - t >= 0 && j - p >= 0) dp[i][j][1] = std::max(dp[i][j][1], dp[i - t][j - p][0]); } if(s[i - 1] == '3') { dp[i][j][0] = std::max(dp[i][j][0], dp[i - 1][j][0] + 1); if(i - t >= 0 && j - p >= 0) dp[i][j][0] = std::max(dp[i][j][0], dp[i - t][j - p][1]); dp[i][j][1] = std::max(dp[i][j][1], dp[i - 1][j][1]); if(i - t >= 0 && j - p >= 0) dp[i][j][1] = std::max(dp[i][j][1], dp[i - t][j - p][0]); } } } answer = dp[n][k][0]; if(answer < 0) answer = -1; std::cout << answer << '\n'; } |
English