牛客练习赛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
#include
#include
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;
} 

你可能感兴趣的:(牛客网,动态规划)