洛谷 P2331 最大子矩阵 题解

题面

对于m==1和m==2两种状态进行不同的dp；

设sum[i][1]表示第一列的前缀和，sum[i][2]表示第二列的前缀和；

sum[i][1]=sum[i-1][1]+a[i][1];

sum[i][2]=sum[i-1][2]+a[i][2];

当m=1时，

设f[i][j]表示前i个数选择j个区间所得到的最大值；

那么枚举0<=p<i, f[i][j]=max(f[i-1][j],f[p][j-1]+sum[i][1]-sum[p][1]);

答案就是f[n][k];

当m=2时，与上面的类似；

f[i][j][p]表示第一列选择前i个数，第二列选择前j个数，总共选择p个矩阵的最大值；

那么枚举0<=p<i,f[i][j][p]=max(f[i-1][j][p],f[i][j-1][p],f[i][j][p],f[l][j][p-1]+sum[i][1]-sum[l][1],f[i][l][p-1]+sum[j][2]-sum[l][2],f[l][l][p-1]+sum[i][2]+sum[i][1]-sum[l][2]-sum[l][1]);

答案就是f[n][n][k];

#include <bits/stdc++.h>
using namespace std;
int n,m,K;
int a[][];
int sum[][];
int g[][];
void solve1()
{
    for(int k=;k<=K;k++){
        for(int i=;i<=n;i++){
            g[i][k]=g[i-][k];
            for(int j=;j<i;j++){
                g[i][k]=max(g[j][k-]+sum[i][]-sum[j][],g[i][k]);
            }
        }
    }
    cout<<g[n][K];
}
int f[][][];
void solve2()
{
    for(int k=;k<=K;k++){
        for(int i=;i<=n;i++){
            for(int j=;j<=n;j++){
                f[i][j][k]=max(f[i-][j][k],f[i][j-][k]);
                for(int l=;l<i;l++){
                    f[i][j][k]=max(f[i][j][k],f[l][j][k-]+sum[i][]-sum[l][]);
                }
                for(int l=;l<j;l++){
                    f[i][j][k]=max(f[i][j][k],f[i][l][k-]+sum[j][]-sum[l][]);
                }
                if(i==j){
                    for(int l=;l<i;l++){
                        f[i][j][k]=max(f[i][j][k],f[l][l][k-]+sum[i][]+sum[i][]-sum[l][]-sum[l][]);
                    }
                }
            }
        }
    }
    cout<<f[n][n][K];
}
int main()
{
    cin>>n>>m>>K;
    for(int i=;i<=n;i++){
        for(int j=;j<=m;j++){
            cin>>a[i][j];
        }
    }
    for(int i=;i<=n;i++){
        sum[i][]=sum[i-][]+a[i][];
        sum[i][]=sum[i-][]+a[i][];
    }
    if(m==){
        solve1();
    }
    else{
        solve2();
    }
}
/*
3 1 2
1
-2
2
*/