牛课练习赛36 Rabbit的工作(1)

链接：https://ac.nowcoder.com/acm/contest/328/B
来源：牛客网
 

题目描述

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

题意很好理解， 但是想的太简单了，暴力模拟以为可以过，少想一个点，只过了80几的样例，但是主要方法还是没想对。

看到大神的解法，可以搜索也可以dp，下面是dp解法。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int Maxn = 1000 + 10 ;

int n, m;
char str[Maxn];
int dp[Maxn][Maxn];

int main() {
    scanf("%d%d%s", &n, &m, str + 1);
    memset(dp, 0x3f3f3f3f, sizeof(dp));
    dp[0][0] = 0;
    //dp[j][k]表示在第i天之前工作j天，并且已经连续工作k天的情况下的最小体力花费
    for (int i = 1; i <= n; i++) {
        for (int j = i; j >= 1; j--) {
            for (int k = j; k >= 1; k--) {
                dp[j][0] = min(dp[j][0], dp[j][k]);
                if (str[i] == '1') {
                    dp[j][k] = min(dp[j][k], dp[j - 1][k - 1] + k);
                }
            }
        }
    }
    int ans = 0;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            if (dp[i][j] <= m) {
                ans = i;
            }
        }
    }
    cout << ans << endl;
}