【DP】LGP1174 打砖块

【题目】
luogu
$n$个栈，每个栈有$m$个元素。元素有黑色和白色，每个元素有一个价值，你可以取出$k$次栈顶的元素，每取出一个黑色元素可以额外获得一次取出机会，求最大价值和。$1\le n,m,k,w_i\le 200$

【解题思路】
一道思博题（并不），然而我一直以为我自己假了，认为枚举前面选了多少个元素是$O(n^2)$级别的，然后就坑了。

首先这道题如果直接按照背包来做是不行的，因为有后效性，可以先取后面的再取前面的，由于如果要取至少要剩下一次机会，因此两者顺序会产生一定差异。

实际上产生差异的过程在于我在这里要不要使用最后一次机会，还是先取了别的再回来取最后一次？

而这种差异具有传递性，即我可以将这次机会一路保留到后面再一路打回来。但这和只考虑后一个没有区别。

那么我们不妨设$f_{i,j,0/1}$表示前$i$个栈，取了$j$次，最后一次是否在这里取。什么意思？因为若最后一次取了一个黑色，我们可以不继续往下取，此时的状态就是$0$。若继续往下取就是$1$，两者的差距在于是否留下一次机会。

那么我们可以根据状态预处理出$g$数组表示对于每个栈来说最后一次是否在这里打得到的价值。

转移时大概有以下三种

• 不在这里取最后一次，最后一次可以用于其他地方取
• 在这里取最后一次，且最后一次是使用之前剩下的次数
• 不在这里取最后一次，但这里剩下的一次次数用于前面取完。

只可意会不可言传，这个方程我改了很多次才拍过交。

复杂度$O(n^3)$

【参考代码】

#include<bits/stdc++.h>
using namespace std;

const int N=205;
int n,m,K;
char ch[3];
int a[N][N],b[N][N],f[N][N][2],g[N][N][2];//f_i,j,k=shot j from 1 to i,use the las on i or not; 1=yes
						  				  //g=only think about this stack,and should use first gun or not
void gmax(int &x,int y){x=max(x,y);}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("LGP1174.in","r",stdin);
	freopen("LGP1174.out","w",stdout);
#endif
	scanf("%d%d%d",&m,&n,&K);//dep m,num n 
	for(int i=1;i<=m;++i) for(int j=1;j<=n;++j) 
	{
		scanf("%d%s",&a[j][m-i+1],ch);
		b[j][m-i+1]=ch[0]=='Y'?1:0;
	}
	for(int i=1;i<=n;++i)
	{
		int cnt=1;
		while(cnt<=m && b[i][cnt]) g[i][0][0]+=a[i][cnt],++cnt;
		for(int j=1;j<=m && cnt<=m;++j)
		{
			g[i][j][0]=g[i][j][1]=g[i][j-1][0]+a[i][cnt];++cnt;
			while(cnt<=m && b[i][cnt]) g[i][j][0]+=a[i][cnt],++cnt;
		}
	}
	for(int i=1;i<=n;++i) for(int j=0;j<=K;++j) for(int k=0;k<=min(m,j);++k)
	{
		gmax(f[i][j][0],f[i-1][j-k][0]+g[i][k][0]);//shot can be use next time
		if(k<j) gmax(f[i][j][1],f[i-1][j-k][1]+g[i][k][0]);//shot last not here but use all gun,so wrong because I type k<i
		if(k) gmax(f[i][j][1],f[i-1][j-k][0]+g[i][k][1]);//shot last here and use a gun before
	}
	printf("%d\n",f[n][K][1]);
	return 0;
}