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

链接：https://ac.nowcoder.com/acm/problem/21675
来源：牛客网
 

Rabbit的工作(1)

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld

题目描述

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

当 s[i]==’1’时可以选择工作或者休息，当 s[i]==’0’时只能休息

我们用 dp[i][j][k]表示考虑到第 i 天，当前已经一共工作了 j 天，且现在连续工作的天数为 k 的时候剩余体力的最大值。

那么我们就可以根据当前位是 0 还是 1 写出状态转移方程

当 s[i]==’0’时 dp[i][j][0]=max(dp[i][j][0],dp[i-1][j][k])

当 s[i]==’1’时 dp[i][j][0]=max(dp[i][j][0],dp[i-1][j][k]) (不工作)

                     dp[i][j][k+1]=max(dp[i][j][k+1],dp[i-1][j][k]-k-1) (工作)

最后找到满足 dp[n][j][k]>=0 最小的 j 即可。

注意这道题不能想当然地贪心(把当前连续最长的 1 最中间的变为 0)，如下面的样例 912 111111111

#include <bits/stdc++.h>
using namespace std;
#define LL long long
#define MAXN 401
int dp[2][MAXN][MAXN];

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    char a[1005];
    scanf("%s",a);
    for(int i=0;i<2;i++){
        for(int j=0;j<MAXN;j++){
            for(int k=0;k<MAXN;k++)
                dp[i][j][k]=INT_MAX/2;
        }
    }
    dp[0][0][0]=0;
    int now=1;
    for(int i=1;i<=n;i++){
        if(a[i-1]=='0'){
            for(int j=0;j<i;j++){
                for(int k=0;k<=j;k++){
                    dp[now][j][0]=min(dp[now^1][j][k],dp[now][j][0]);
                }
            }
        }
        else {
            for(int j=0;j<i;j++){
                for(int k=0;k<=j;k++){
                    dp[now][j][0]=min(dp[now^1][j][k],dp[now][j][0]);
                }
            }
            for(int j=1;j<=i;j++){
                for(int k=1;k<=j;k++){
                    dp[now][j][k]=min(dp[now^1][j-1][k-1]+k,dp[now][j][k]);
                }
            }
        }
        now=now^1;
    }
    int re=INT_MIN;
    for(int i=0;i<2;i++){
        for(int j=0;j<MAXN;j++){
            for(int k=0;k<MAXN;k++)
                if(dp[i][j][k]<=m){re=max(re,j);}
        }
    }
    printf("%d",re);
    return 0;
}