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

题目描述

Rabbit大学毕业后找到了一份实习工作，如果实习通过她就转正了。
实习期共有N天，其中有几天公司集体放假，Rabbit不用上班，剩下时间她可以选择工作或者休息。Rabbit工作总是越来越累，可是每当她休息时，她就重新充满了能量。简而言之，Rabbit第一天工作时这一天会消耗体力1，连续第二天工作时这一天会消耗体力2，连续第三题工作时这一天会消耗体力3，以此类推…每当她休息后，工作的第一天又会消耗体力1。
为了让boss满意，Rabbit想工作尽量多的天数，但是懒惰的Rabbit又想让自己的总体力消耗不超过K。
输入描述:
第一行两个整数N，K。
第二行一个长度为N的01字符串。如果第i个字符为‘1’，表示这一天Rabbit可以选择工作或者休息，否则这一天Rabbit放假。
输出描述:
输出Rabbit最多能工作的天数。
输入
4 2
1011
输出
2
说明
第三天和第四天里面休息一天即可，总体力消耗为2
备注:
1<=N<=400
1<=K<=N∗(N+1)/2

code

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 405;
int main(){
	int n, m, dp[2][maxn][maxn];//dp[i][j][k]=前i天工作了j天且连续工作k天的最大剩余体力 
	char s[maxn];
	scanf("%d%d%s", &n, &m, s+1);
	memset(dp, -1, sizeof(dp));
	int now = 0;
	dp[now][0][0] = m; 
	for(int i = 1; i <= n; i++){
		memset(dp[now^1], -1, sizeof(dp[now^1]));
		for(int j = 0; j <= n; j++){
			for(int k = 0; k <= n; k++){
				if(dp[now][j][k]<0) continue;
				if(s[i]=='0') dp[now^1][j][0] = max(dp[now^1][j][0], dp[now][j][k]);
				else{
					dp[now^1][j][0] = max(dp[now^1][j][0], dp[now][j][k]);
					dp[now^1][j+1][k+1] = max(dp[now^1][j+1][k+1], dp[now][j][k]-k-1);
				}
			}
		}
		now^=1;
	}
	for(int i = n; i >= 0; i--){
		for(int j = 0; j <= i; j++){
			if(dp[now][i][j]>=0){
				printf("%d\n", i);
				return 0;
			} 
		}
	}
	return 0;
}