Rabbit的工作(1)

思路:对于是0的位置直接选择休息,对于是1的位置要么工作,要么休息,二选一的问题,有点像01背包的选或者不选的问题,所以考虑效仿01背包。答案是要求最多共工作了几天,那么开一个维度表示共工作了几天,因为此题跟连续工作的天数有关,所以我们需要加一个维度表示目前为止连续工作的天数。
那么也就是三个维度 dp[i][j][k]表示对于前i天一共工作了j天目前已经连续工作了k天的最小体力花费
此题空间给了64M 开三维会MLE 因为很像01背包 所以考虑滚动数组优化掉第一个维度
然后注意对于1的位置是选或者不选 也就是第二维度这里 我们应该倒序遍历,否则本来每个1只能最多工作一天 这样会导致工作很多天 不太明白的考虑一下01背包的倒序遍历即可
那么最后只要找dp[j][k]≤k的最大的j即为所求答案

#include<bits/stdc++.h>
using namespace std;
int dp[505][505];
char s[505];
int main()
{
    int n,k;
    cin>>n>>k;
    cin>>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++){
                if(s[i]=='0'){///必须休息
                    dp[j][0]=min(dp[j][0],dp[j][k]);
                }
                else///可以工作可以休息
                {
                    dp[j][0]=min(dp[j][0],dp[j][k]);///选择休息
                    dp[j][k]=min(dp[j][k],dp[j-1][k-1]+k);///选择工作
                }
            }
        }
    }
    int m=0;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=i;j++){
            if(dp[i][j]<=k) m=i;
        }
    }
    cout<<m<<endl;
    return 0;
}