Bzoj P1084 [SCOI2005]最大子矩阵___动态规划

题目大意：

有一个$n\ast m$的矩阵，请你选出其中$K$个子矩阵，使得这个$K$个子矩阵分值之和最大，选出的$K$个子矩阵不能相互重叠。
$K$个子矩阵分值之和最大为多少
$1\le n\le 100,1\le m\le 2,1\le K\le 10$

分析：

注意到$1\le m\le 2$，
分类讨论一下，
①$m=1$，是一个$n\ast 1$的矩阵，把它看成一个长度为$n$的序列
设$f_{i,j}$表示前$i$个数选了$j$个区间的最大分值之和
设$Su{m}_{i,j}$表示区间$[i,j]$的分值之和，前缀和搞一下
转移显然：
初值$f_{i,j}=f_{i-1,j}$，
转移时枚举一下最后一个区间是怎样的：
$f_{i,j}=max(f_{i,j},f_{k,j-1}+Su{m}_{k+1,i})$
$Answer=f_{n,K}$
②$m=2$，是一个$n\ast 2$的矩阵，把它看成一个长度为$n$，有2层的序列
设$f_{i,j,k}$表示第一层前$i$个第二层前$j$个选了$k$个子矩阵的最大分值之和
设$Su{m}_{k,i,j}$第$k$层的区间$[i,j]$的分值之和
转移显然：
初值$f_{i,j,k}=max(f_{i-1,j,k},f_{i,j-1,k})$
转移时枚举一下最后一个矩阵是怎样的，
当$i!=j$时，最后一个选的子矩阵应该是$x\ast 1$的，
即
$f_{i,j,k}=max(f_{i,j,k},f_{l-1,j,k-1}+Su{m}_{1,l,i})$
$f_{i,j,k}=max(f_{i,j,k},f_{i,l-1,k-1}+Su{m}_{2,l,j})$
特殊的，当$i=j$时，除了上述$2$个情况以外，最后一个选的矩阵也可以是$x\ast 2$的，
即
$f_{i,j,k}=max(f_{i,j,k},f_{l-1,l-1,k-1}+Sum1,l,i+Su{m}_{2,l,j})$
$Answer=f_{n,n,K}$

然后注意一下初值要赋一下一个较大的负数，否则可能会WA

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <cstdlib>
#include <algorithm>
 
#define inf 0x3f3f3f3f
#define N 105
#define M 15
 
using namespace std;
 
int Bdp[N][N][M], Adp[N][M], Sum[3][N], n, m, K;
 
void WorkA()
{
    for (int i = 0; i <= n; i++) 
        for (int j = 1; j <= K; j++) Adp[i][j] = -inf;
         
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= K; j++) 
        {
            Adp[i][j] = Adp[i - 1][j];
            for (int k = 1; k <= i; k++)
                Adp[i][j] = max(Adp[i][j], Adp[k - 1][j - 1] + (Sum[1][i] - Sum[1][k - 1]));
        }
    printf("%d\n", Adp[n][K]);
}
 
void WorkB()
{
    for (int i = 0; i <= n; i++)
        for (int j = 0; j <= n; j++) 
            for (int k = 1; k <= K; k++) Bdp[i][j][k] = -inf;
             
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            for (int k = 1; k <= K; k++)
            {  
                Bdp[i][j][k] = max(Bdp[i - 1][j][k], Bdp[i][j - 1][k]);        
                for (int l = 1; l <= i; l++)
                    Bdp[i][j][k] = max(Bdp[i][j][k], Bdp[l - 1][j][k - 1] + (Sum[1][i] - Sum[1][l - 1]));
                for (int l = 1; l <= j; l++)
                    Bdp[i][j][k] = max(Bdp[i][j][k], Bdp[i][l - 1][k - 1] + (Sum[2][j] - Sum[2][l - 1]));
                if (i == j)
                {
                    for (int l = 1; l <= i; l++)
                        Bdp[i][j][k] = max(Bdp[i][j][k], Bdp[l - 1][l - 1][k - 1] + (Sum[1][i] - Sum[1][l - 1]) + (Sum[2][i] - Sum[2][l - 1]));    
                }       
            }   
    printf("%d\n", Bdp[n][n][K]);
}
 
int main()
{
    scanf("%d %d %d", &n, &m, &K);
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++) 
        {
            int x; scanf("%d", &x);
            Sum[j][i] = Sum[j][i - 1] + x;
        }
    if (m == 1) WorkA(); else WorkB();
    return 0;
}