最大子矩阵 DP

最大子矩阵

Time Limit:10000MS Memory Limit:165888KB 64bit IO Format:%lld & %llu

HYSBZ 1084

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
3 2 2
1 -3
2 3
-2 3
Sample Output
9

题解：

虽然一看就是DP，可是一开始完全不知道如何DP，后来看到了数据范围…笑了，当m=1的时候就是普通的最大连续子串和，麻烦的是当m=2的时候的状态转移很麻烦，我们设f[i][j][k]为第一列选前i个，第二列选前j个，选了k个时候的最大权值
那么有
f[i][j][k] = max(f[i-1][j][k],f[i][j-1][l]);
f[i][j][k] = max(f[ii][j][k-1]+sum[ii+1][i][0]) 重新调整第一列
f[i][j][k]=max(f[i][jj][k-1]+sum[jj+1][j][1]) 重新调整第二列
当i==j时就是形成一个矩阵，再处理就是了

代码

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <string>
#include <set>
#include <ctime>
#include <cmath>
#include <cctype>
using namespace std;
#define LL long long
int cas=1,T;

const int maxn = 100;
const int maxm =20;

using namespace std;

int n,m,k;
int a[maxn][maxn],sum[maxn][maxn],f[maxn][maxm],ff[maxn][maxn][maxm];

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    sum[0][1]=sum[0][2]=0;
    for (int i=1;i<=n;i++) 
        for (int j=1;j<=m;j++) 
            scanf("%d",&a[i][j]),sum[i][j]=sum[i-1][j]+a[i][j];
    if (m==1)
    {
        memset(f,0,sizeof(f));
        for (int i=1;i<=n;i++) 
        {
            f[i][0]=0;
            for (int l=1;l<=k;l++)
            {
                f[i][l]=f[i-1][l];
                for (int j=0;j<i;j++)
                {
                    f[i][l]=max(f[i][l],f[j][l-1]+sum[i][1]-sum[j][1]);
                }
            }
        }
        printf("%d\n",f[n][k]);
    }
    else 
    {
        memset(ff,0,sizeof(ff));
        for (int i=1;i<=n;i++)
        {
            for (int j=1;j<=n;j++)
            {
                ff[i][j][0]=0;
                for (int l=1;l<=k;l++)
                {
                    ff[i][j][l]=max(ff[i-1][j][l],ff[i][j-1][l]);
                    for (int ii=0;ii<i;ii++) 
                        ff[i][j][l]=max(ff[i][j][l],ff[ii][j][l-1]+sum[i][1]-sum[ii][1]);
                    for (int jj=0;jj<j;jj++) 
                        ff[i][j][l]=max(ff[i][j][l],ff[i][jj][l-1]+sum[j][2]-sum[jj][2]);
                    if (i==j)
                        for (int jj = 0;jj<i;jj++)
                            ff[i][i][l]=max(ff[i][i][l],ff[jj][jj][l-1]+sum[i][1]-sum[jj][1]+sum[j][2]-sum[jj][2]);
                }
            }       
        }
        printf("%d\n",ff[n][n][k]);
    }
    return 0;
}