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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_39562952/article/details/85849983

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;
}

猜你喜欢

转载自blog.csdn.net/qq_39562952/article/details/85849983