C、Rabbit的工作(1)

参考题解7QQQQQQQ orz

一开始我打算用函数对字符串进行从0分隔,然后对每连续的1,一直工作到不能工作在休息,然后重新工作。
但是WA的理所当然……那没办法阿,只能想想别的思路了


正解:动态规划

我们表示前i天工作了j天,现在连续工作k天的最小体力花费。
那么对于后面每新加一天,可以选择工作或者休息,如果休息,就是取到连续工作天中取个最小康康要不要更新即可
如果选择工作,也是前天工作了再把连续工作都挨个更新一遍
最终找到第一个小于等于即可
这里可以因为之后无意义的向后递增1,值不再改变,可以直接省略或者滚动数组都可。

#include <bits/stdc++.h>
#pragma GCC optimize(2)
#pragma GCC optimize(3)
using namespace std;
#define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
typedef long long ll;
const ll MOD = 1e9 + 7;
inline ll read() { ll s = 0, w = 1; char ch = getchar(); while (ch < 48 || ch > 57) { if (ch == '-') w = -1; ch = getchar(); }    while (ch >= 48 && ch <= 57) s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();    return s * w; }
inline void write(ll x) { if (!x) { putchar('0'); return; } char F[200]; ll tmp = x > 0 ? x : -x; if (x < 0)putchar('-');    int cnt = 0;    while (tmp > 0) { F[cnt++] = tmp % 10 + '0';        tmp /= 10; }    while (cnt > 0)putchar(F[--cnt]); }
inline ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; }
ll qpow(ll a, ll b) { ll ans = 1;    while (b) { if (b & 1)    ans *= a;        b >>= 1;        a *= a; }    return ans; }    ll qpow(ll a, ll b, ll mod) { ll ans = 1; while (b) { if (b & 1)(ans *= a) %= mod; b >>= 1; (a *= a) %= mod; }return ans % mod; }
inline int lowbit(int x) { return x & (-x); }

const int N = 405;
int dp[N][N];
char s[N];

int main() {
    int n = read(), m = read();
    scanf("%s", s + 1);
    memset(dp, 0x3f, sizeof(dp));
    dp[0][0] = 0;
    for (int i = 1; i <= n; ++i)
        for (int j = i; j; --j)
            for (int k = 1; k <= j; ++k) {
                dp[j][0] = min(dp[j][0], dp[j][k]);
                if (s[i] == '1')
                    dp[j][k] = min(dp[j][k], dp[j - 1][k - 1] + k);
            }
    for (int j = n; ~j; --j)
        for (int k = 0; k <= j; ++k)
            if (dp[j][k] <= m) {
                printf("%d", j);
                return 0;
            }
    return 0;
}