SCOI2005最大子矩阵和

题目：https://ac.nowcoder.com/acm/problem/20242

链接：https://ac.nowcoder.com/acm/problem/20242
来源：牛客网

题目描述
这里有一个n*m的矩阵，请你选出其中k个子矩阵，使得这个k个子矩阵分值之和最大。
注意：选出的k个子矩阵 不能相互重叠。
输入描述:
第一行为n,m,k（1 ≤ n ≤ 100,1 ≤ m ≤ 2,1 ≤ k ≤ 10），
接下来n行描述矩阵每行中的每个元素的分值(每个元素的分值的绝对值不超过32767)。
输出描述:
只有一行为k个子矩阵分值之和最大为多少。
示例1
输入
复制
3 2 2
1 -3
2 3
-2 3
输出
复制
9

解题思路：
注意到m=1或者2
当m=1时，是普通的最大连续字段和，只不过是k个：
设dp[i][j]表示前j个数中取出i个矩形的最大和
dp方程

dp[i][j]=max(dp[i][j],dp[i-1][k]+sum[j]-sum[k]);
当m=2时，设f[i][j][k]表示第一列选到第i个数，第二列选到第j个数时，总共k个子矩形的答案

转移有4种情况

当这一位什么都不做的时候

dp[i][j][k]=max(dp[i-1][j][k],dp[i][j-1][k])
当仅选取第一列的某段区间时

dp[i][j][k]=max(dp[l][j][k-1]+sum[i][1]-sum[l-1][1])
当仅选取第二列的某段区间时

dp[i][j][k]=max(dp[i][l][k-1]+sum[j][2]-sum[l-1][2])
当i==j时，可以选取两列一起的

dp[i][j][k]=max(f[l][l][k]+sum[i][1]+sum[i][2]-sum[l-1][1]-sum[l-1][2])

以上解题思路来自牛客，弱鸡写不出

#include <bits/stdc++.h>
using namespace std;

int dp[101][101][11];
void solve(){

    int n,m,k;
    cin >> n >> m >> k;

    if(m == 1){

        int dp[11][101],sum[101] = {0},value;
        memset(dp,0,sizeof(dp));
        for(int i = 1;i<=n;i++) {cin >> value;sum[i] = sum[i-1] + value;}
        for(int i = 1;i<=k;i++) //子矩阵个数
            for(int j = 1;j<=n;j++){    //元素个数
                dp[i][j] = dp[i][j-1];
                for(int k = j-1;k>=0;k--)
                    dp[i][j] = max(dp[i][j],dp[i-1][k] + sum[j] - sum[k]);
            }
        cout << dp[k][n];
    }else{
        //sum1表示第一列前缀和，sum2表示第二列前缀和
        int sum1[101]={0},sum2[101]={0},value1,value2;
        for(int i = 1;i<=n;i++) {cin >> value1 >> value2; sum1[i] = sum1[i-1] + value1; sum2[i] = sum2[i-1] + value2;}
        int kk = k;
        for(int k = 1;k<=kk;k++)
            for(int i = 1;i<=n;i++)
                for(int j = 1;j<=n;j++){
                    dp[i][j][k] = max(dp[i-1][j][k],dp[i][j-1][k]);
                    for(int l = i-1;l>=0;l--) dp[i][j][k] = max(dp[i][j][k],dp[l][j][k-1] + sum1[i] - sum1[l]);
                    for(int r = j-1;r>=0;r--) dp[i][j][k] = max(dp[i][j][k],dp[i][r][k-1] + sum2[j] - sum2[r]);
                    if(i == j)
                        for(int l = i-1;l>=0;l--) dp[i][j][k] = max(dp[i][j][k],dp[l][l][k-1]+sum1[i]-sum1[l]+sum2[j]-sum2[l]);
                    
                }
        cout << dp[n][n][k];
    }
}

int main(){

    ios_base::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);

    solve();

    return 0;
}