题目描述
传送门
题目大意:给出一个n*m的矩阵,请你选出其中至多k个子矩阵,使得这个k个子矩阵分值之和最大。注意:选出的k个子矩阵不能相互重叠。
题解
m只有1,2两种,所以分开考虑一下
当m=1时,
f[i][j]
表示到i选取了j段的最大价值
f[i][j]=max(f[i−1][j],f[k][j−1]+sum[i]−sum[k])
当m=2时,
f[i][j][0..4]
表示到第i行选择了j个子矩阵,第i行的状态为0..4的最大价值
其实是一个类似状压的DP。
0表示第i行的两个点都不选
1表示只选第一个点,2表示只选第二个点
3表示两个点都选但不属于同一个子矩形
4表示两个点都选但属于同一个子矩形
然后转移的时候考虑和上一行的情况怎么合并即可。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 103
using namespace std;
int f[N][15][5],n,m,k;
int a[N][3],dp[N][N],sum[N];
int main()
{
freopen("a.in","r",stdin);
freopen("my.out","w",stdout);
scanf("%d%d%d",&n,&m,&k); int ans=0;
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++) sum[i]=sum[i-1]+a[i][1];
memset(dp,0,sizeof(dp));
for (int i=0;i<=n;i++) dp[i][0]=0;
for (int j=1;j<=k;j++)
for (int i=1;i<=n;i++) {
dp[i][j]=max(dp[i][j],dp[i-1][j]);
for (int t=0;t<i;t++)
dp[i][j]=max(dp[i][j],dp[t][j-1]+sum[i]-sum[t]);
ans=max(ans,dp[i][j]);
}
printf("%d\n",ans);
return 0;
}
memset(f,128,sizeof(f)); int INF=f[0][0][0];
f[1][0][0]=0; f[1][1][1]=a[1][1]; f[1][1][2]=a[1][2];
f[1][2][3]=f[1][1][4]=a[1][1]+a[1][2];
for (int i=2;i<=n;i++)
for (int j=1;j<=k;j++) {
int mx=INF; int mx1=INF; int mx2=INF;
for (int l=0;l<=4;l++) {
mx=max(mx,f[i-1][j][l]);
mx1=max(mx1,f[i-1][j-1][l]);
if (j-2>=0) mx2=max(mx2,f[i-1][j-2][l]);
f[i][0][0]=0;
}
f[i][j][0]=mx; int aa=a[i][1]+a[i][2];
f[i][j][1]=max(f[i][j][1],mx1+a[i][1]);
f[i][j][1]=max(f[i][j][1],max(f[i-1][j][1],f[i-1][j][3])+a[i][1]);
f[i][j][2]=max(f[i][j][2],mx1+a[i][2]);
f[i][j][2]=max(f[i][j][2],max(f[i-1][j][2],f[i-1][j][3])+a[i][2]);
f[i][j][3]=max(f[i][j][3],mx2+aa);
f[i][j][3]=max(f[i][j][3],max(f[i-1][j-1][1],f[i-1][j-1][2])+aa);
f[i][j][3]=max(f[i][j][3],f[i-1][j][3]+aa);
f[i][j][4]=max(f[i][j][4],mx1+aa);
f[i][j][4]=max(f[i][j][4],f[i-1][j][4]+aa);
}
for (int i=1;i<=n;i++)
for (int j=1;j<=k;j++)
for (int l=0;l<=4;l++) ans=max(ans,f[i][j][l]);
printf("%d\n",ans);
}