BZOJ 1084 最大子矩阵 线性dp

给出一个$n\times m$的矩阵，其中$n\le 100,m\le 2$，然后求选取$k$个子矩阵的最大值是多少。注意到这个$m$只有$1$和$2$两种情形。
当$m=1$的时候，这个问题退化为一个长度为$n$的序列选$k$段的最大值。$f_{i,j}$表示前$i$个中选择$j$段的最大值。对于不选的情形：$f_{i,j}=f_{i-1,j}$，对于选的情形，枚举当前这一段的左端点。 f i , j = max ⁡ { f k , j − 1 + s u m k + 1 , i } f_{i,j}=\max\{f_{k,j-1}+sum_{k+1,i}\} fi,j​=max{fk,j−1​+sumk+1,i​}
推广$m=2$的情形，相当于两个长度为$n$的序列选$k$段的最大值。同理不选的情形： f i , j , k = max ⁡ { f i − 1 , j , k , , f i , j − 1 , k } f_{i,j,k}=\max\{{f_{i-1,j,k,},f_{i,j-1,k}\}} fi,j,k​=max{fi−1,j,k,​,fi,j−1,k​}，选择的话就分别对两个序列处理，同时加上两个长度相等的情形。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=1e2+3; 
int n,m,K;
namespace solve_one {
	int a[N],sum[N],f[N][N];
	void read() {
		for(int i=1;i<=n;i++) {
			scanf("%d",&a[i]);
			sum[i]=sum[i-1]+a[i];
		}
	}
	void solve() {
		for(int k=1;k<=K;k++) {
			for(int i=1;i<=n;i++) {
				f[i][k]=f[i-1][k];
				for(int j=1;j<=i;j++) {
					f[i][k]=max(f[i][k],f[j-1][k-1]+(sum[i]-sum[j-1]));
				}
			}
		}
		printf("%d\n",f[n][K]);
	}
}
namespace solve_two {
	int a[N][3];
	int suml[N],sumr[N];
	int f[N][N][11];
	void read() {
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=2;j++) 
				scanf("%d",&a[i][j]);
			suml[i]=suml[i-1]+a[i][1];
			sumr[i]=sumr[i-1]+a[i][2];
		}
	}
	void solve() {
		for(int k=1;k<=K;k++) {
			for(int l=1;l<=n;l++) {
				for(int r=1;r<=n;r++) {
					f[l][r][k]=max(f[l-1][r][k],f[l][r-1][k]);
					for(int j=1;j<=l;j++) f[l][r][k]=max(f[l][r][k],f[j-1][r][k-1]+suml[l]-suml[j-1]);
					for(int j=1;j<=r;j++) f[l][r][k]=max(f[l][r][k],f[l][j-1][k-1]+sumr[r]-sumr[j-1]);
					if(l==r) for(int j=1;j<=l;j++) f[l][r][k]=max(f[l][r][k],f[j-1][j-1][k-1]+suml[l]+sumr[r]-suml[j-1]-sumr[j-1]); 	
				}
			}
		}
		printf("%d\n",f[n][n][K]); 
	}
}
int main() {
	scanf("%d%d%d",&n,&m,&K);
	if(m==1) {
		solve_one::read();
		solve_one::solve(); 
	}
	else if(m==2) {
		solve_two::read();
		solve_two::solve();
	}
	return 0;
}