1084: [SCOI2005]最大子矩阵
Time Limit: 10 Sec Memory Limit: 162 MBSubmit: 2772 Solved: 1380
[ Submit][ Status][ Discuss]
Description
这里有一个n*m的矩阵,请你选出其中k个子矩阵,使得这个k个子矩阵分值之和最大。注意:选出的k个子矩阵不能相互重叠。
Input
第一行为n,m,k(1≤n≤100,1≤m≤2,1≤k≤10),接下来n行描述矩阵每行中的每个元素的分值(每个元素的
分值的绝对值不超过32767)。
Output
只有一行为k个子矩阵分值之和最大为多少。
Sample Input
Sample Output
当m=1的时候就是个简单的k个最大子段和就不说了
当m=2时
dp[i][j][k]表示第1列取到前i个数,第2列取到前j个数,共用了k个矩阵所得到的最大值
有④种情况
①什么都不做:dp[i][j][k] = max(dp[i-1][j][k], dp[i][j-1][k])
②将第1列第i个数包在某个宽为1的矩形内:
dp[i][j][k] = max(dp[c][j][k-1]+sum[i][1]-sum[c][1], dp[i][j][k]) (1<=c<=i-1表示矩形的头)
③将第2列第j个数包在某个宽为1的矩形内:
dp[i][j][k] = max(dp[i][c][k-1]+sum[j][2]-sum[c][2], dp[i][j][k]) (1<=c<=j-1表示矩形的头)
④当i==j时,将第1列和第2列的第i(j)个数包在某个宽为2的矩形内:
dp[i][j][k] = max(dp[c][c][k-1]+sum[i][1]-sum[c][1]+sum[j][2]-sum[c][2], dp[i][j][k])
其中sum[][]表示前缀和
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[105][2], sum[105][2], dp[105][105][15];
int main(void)
{
int n, m, i, j, p, k, ans, c;
while(scanf("%d%d%d", &n, &m, &p)!=EOF)
{
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
scanf("%d", &a[i][j]);
sum[i][j] = sum[i-1][j]+a[i][j];
}
}
memset(dp, -55, sizeof(dp));
ans = -2147483647;
for(i=0;i<=n;i++)
{
for(j=0;j<=n;j++)
dp[i][j][0] = 0;
}
if(m==1)
{
for(i=1;i<=n;i++)
{
for(j=1;j<=p;j++)
{
dp[i][0][j] = max(dp[i][0][j], dp[i-1][0][j]);
for(k=0;k<=i-1;k++)
dp[i][0][j] = max(dp[k][0][j-1]+sum[i][1]-sum[k][1], dp[i][0][j]);
ans = max(ans, dp[i][0][j]);
}
}
}
else
{
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
for(k=1;k<=p;k++)
{
dp[i][j][k] = max(dp[i-1][j][k], dp[i][j-1][k]);
for(c=0;c<=i-1;c++)
dp[i][j][k] = max(dp[c][j][k-1]+sum[i][1]-sum[c][1], dp[i][j][k]);
for(c=0;c<=j-1;c++)
dp[i][j][k] = max(dp[i][c][k-1]+sum[j][2]-sum[c][2], dp[i][j][k]);
if(i==j)
{
for(c=0;c<=i-1;c++)
dp[i][j][k] = max(dp[c][c][k-1]+sum[i][1]-sum[c][1]+sum[j][2]-sum[c][2], dp[i][j][k]);
}
ans = max(ans, dp[i][j][k]);
}
}
}
}
printf("%d\n", ans);
}
return 0;
}