dp-牛客算法周周练7-C Rabbit的工作(1)

题目链接
题目描述
Rabbit大学毕业后找到了一份实习工作，如果实习通过她就转正了。
实习期共有N天，其中有几天公司集体放假，Rabbit不用上班，剩下时间她可以选择工作或者休息。Rabbit工作总是越来越累，可是每当她休息时，她就重新充满了能量。简而言之，Rabbit第一天工作时这一天会消耗体力1，连续第二天工作时这一天会消耗体力2，连续第三天工作时这一天会消耗体力3，以此类推…每当她休息后，工作的第一天又会消耗体力1。
为了让boss满意，Rabbit想工作尽量多的天数，但是懒惰的Rabbit又想让自己的总体力消耗不超过K。
输入描述:
第一行两个整数N，K。
第二行一个长度为N的01字符串。如果第i个字符为‘1’，表示这一天Rabbit可以选择工作或者休息，否则这一天Rabbit放假。
输出描述:
输出Rabbit最多能工作的天数。
示例1
输入
4 2
1011
输出
2
说明
第三天和第四天里面休息一天即可，总体力消耗为2
备注:
1<=N<=400
1<=K<=N∗(N+1)/2
时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld
思路：
定义dp[2][405][405];
dp[1][j][k]表示 1~i天 工作了j天 第i天时连续工作了k天
dp[0][j][k]表示 1~i-1天 工作了j天 第i-1天时连续工作了k天
要用1和0减少内存，否则会内存超限
代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
char s[405];
ll dp[2][405][405]; //dp[1][j][k]表示 1~i天 工作了j天 第i天时连续工作了k天   dp[0][j][k]表示 1~i-1天 工作了j天 第i-1天时连续工作了k天
int main()
{
    ll n, m, i, j, k;
    scanf("%lld%lld", &n, &m);
    scanf("%s", s + 1);
    for (i = 0; i <= 1; i++)
        for (j = 0; j <= n; j++)
            for (k = 0; k <= n; k++)
                dp[i][j][k] = 1e8;
    dp[0][0][0] = 0;
    ll sum = 0;
    ll ans = 0;
    for (i = 1; i <= n; i++)
    {
        for (j = 0; j <= n; j++)
        {
            if (s[i] == '0')
            {
                for (k = 0; k <= n; k++)
                    dp[1][j][0] = min(dp[1][j][0], dp[0][j][k]);
            }
            else
            {
                for (k = 0; k <= n; k++)
                {
                    dp[1][j][0] = min(dp[1][j][0], dp[0][j][k]);
                    if (k > 0 && j > 0)
                        dp[1][j][k] = min(dp[1][j][k], dp[0][j - 1][k - 1] + k);
                }
            }
        }
        for (j = 0; j <= n; j++)
        {
            for (k = 0; k <= n; k++)
            {
                if (dp[1][j][k] <= m)
                    ans = max(ans, j);
                dp[0][j][k] = dp[1][j][k];
                dp[1][j][k] = 1e8;
            }
        }
    }
    printf("%lld\n", ans);
}