牛客训练赛36 B题(dp滚动数组)

2019年第一次比赛，遭逢大败。。。都不忍心看自己的排名，第一题一直在想该怎么去优化。发现并不用，直接暴力过。。。心态炸了，第二题是dp题，一开始想到了增加维度以获得更多信息，还好死不死的用疲劳度作为第二维。。。结果数组变成400*81000那么大。。。一直在推状态转移方程还推不出来。。。看了题解恍然大悟。。。哎，在推不出状态转移方程的时候应该改变思路，想办法减小每一维的大小，这样就比较容易想到正确的办法，具体看题解；

这题还有一个坑点就是如果不用滚动数组，会爆内存。。。

 

代码如下：

/*
想到了扩充维度来获取更多信息。但是把疲劳度作为了第二维。。。所以一直转移不出来 
dp[i][j][k]:第i天，已经工作了j天，已经连续工作了k天，剩余的疲劳度 
ans=max(j); 
滚动数组 
*/
#define LOCAL
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
#include<cmath>
using namespace std;
#define maxn 2100
#define inf 40000000
typedef pair<int,int >P;
int d[2][410][410];//每个状态下的最大收益 
int n,m;
char s[410];
int main()
{
	#ifdef LOCAL
	//freopen("data.in","r",stdin);
	//freopen("data.out","w",stdout);
	#endif
	scanf("%d%d",&n,&m);
	scanf("%s",s+1);
	int maxx=0;
	int ans=0;
	memset(d,-1,sizeof(d));
	d[0][0][0]=m;
	int cur=1;
	for(int i=1;i<=n;i++)
	{
		int j;
		for(j=0;j<=n;j++)
		{
			for(int k=0;k<=n;k++)
			{
				d[cur][j][0]=max(d[cur][j][0],d[cur^1][j][k]);
				
			}
		}
		if(d[cur][j][0]>=0)
			ans=max(ans,j);
		if(s[i]=='1')
		{
			for(int j=1;j<=n;j++)
			{
				for(int k=1;k<=n;k++)
				{
					d[cur][j][k]=max(d[cur][j][k],d[cur^1][j-1][k-1]-k);
					if(d[cur][j][k]>=0)
						ans=max(ans,j);
				}
			}	
				
		}
		cur^=1;
	}
	
	printf("%d\n",ans);
	return 0;
}