luogu P2331 [SCOI2005]最大子矩阵

题目传送门：https://www.luogu.org/problemnew/show/P2331

题意：

有一个n*m的矩阵，请你选出其中k个子矩阵，使得这个k个子矩阵分值之和最大。注意：选出的k个子矩阵不能相互重叠。

思路：

分析数据范围可知m=1或2，可以判断m，对于不同的m，推出不同的方程（详见代码）。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define INF 2147483647
using namespace std;
	int n,m,k;
	int a[200][5],f[200][20][5];
/*
f[i][j][k]表示选取到第i行,选取了j个矩阵,且这一行属于第k种状态。
对于k 
0:空出这一行
1:选择左边的数
2:选择右边的数
3:两个数都选(但不在同一个矩阵里)
4:两个数都选(在同一个矩阵里)
*/	
int main()
{
	scanf("%d %d %d",&n,&m,&k);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			scanf("%d",&a[i][j]);
	if(m==1)
	{
		for(int i=1;i<=n;i++)
			for(int j=1;j<=k;j++)
			{
				f[i][j][0]=max(f[i-1][j][0],f[i-1][j][1]);
				f[i][j][1]=max(f[i-1][j-1][0],f[i-1][j][1])+a[i][1];
			}
		printf("%d",max(f[n][k][0],f[n][k][1]));
	}
	else
	{
		memset(f,-63,sizeof(f));
		for(int i=0;i<=n;i++)
			for(int j=0;j<=k;j++)
				f[i][j][0]=0;
		for(int i=1;i<=n;i++)
			for(int j=1;j<=k;j++)
			{
				f[i][j][0]=max(f[i-1][j][0],max(f[i-1][j][1],max(f[i-1][j][2],max(f[i-1][j][3],f[i-1][j][4]))));
				f[i][j][1]=max(f[i-1][j-1][0],max(f[i-1][j][1],max(f[i-1][j-1][2],max(f[i-1][j][3],f[i-1][j-1][4]))))+a[i][1];
				f[i][j][2]=max(f[i-1][j-1][0],max(f[i-1][j-1][1],max(f[i-1][j][2],max(f[i-1][j][3],f[i-1][j-1][4]))))+a[i][2];
				f[i][j][3]=max(f[i-1][j-1][1],max(f[i-1][j-1][2],max(f[i-1][j][3],(j>=2?f[i-1][j-2][4]:-INF))))+a[i][1]+a[i][2];
				f[i][j][4]=max(f[i-1][j-1][0],max(f[i-1][j-1][1],max(f[i-1][j-1][2],max(f[i-1][j-1][3],f[i-1][j][4]))))+a[i][1]+a[i][2];
			}
		printf("%d",max(f[n][k][0],max(f[n][k][1],max(f[n][k][2],max(f[n][k][3],f[n][k][4])))));
	}
}