(dp/前缀和) P2331 [SCOI2005]最大子矩阵

https://www.luogu.org/problemnew/show/P2331
这里有一个n*m的矩阵,请你选出其中k个子矩阵,使得这个k个子矩阵分值之和最大。注意:选出的k个子矩阵不能相互重叠。
第一行为n,m,k(1≤n≤100,1≤m≤2,1≤k≤10),接下来n行描述矩阵每行中的每个元素的分值(每个元素的分值的绝对值不超过32767)。
只有一行为k个子矩阵分值之和最大为多少。
in:
3 2 2
1 -3
2 3
-2 3
out:
9

从n * m的矩阵中选出k个子矩阵使总和最大
不知道怎么表示表示这些状态,看了下题解,先用前缀和处理,方便计算
要考虑m<=2这个条件,分别讨论
m=1时就是一列元素选出k个连续部分是总和最大, dp[i][j]表示当前讨论第i个元素,第j个块的值,则状态转移方程即为dp[i][j] = max(dp[i - 1][j], dp[k][j - 1] + sum[i] - sum[k] | k < i);
m=2时dp[i][j][k]表示第一列的第i个元素,第二列的第j个元素,正在处理第k块的值,状态转移为max(dp[i - 1][j][k - 1], dp[i][j - 1][k - 1], dp[x][j][k - 1] | x < i, dp[i][x][k - 1] | x < j, dp[x][y][k - 1] | x < i && y < j);
即为当前的第i个元素和第j个元素分别都不位于第k块,单独位于第k块,一行时都位于第k块的状态的最大值

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

const int maxn = 1e2 + 4;
int s1[maxn], s2[maxn]; //前缀和
int dp[maxn][maxn], f[maxn][maxn][maxn];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n, m, K, x, y;
    cin >> n >> m >> K;
    if(m == 1) { //一列的情况
        for(int i = 1; i <= n; i++)
            cin >> x, s1[i] = s1[i - 1] + x;
        for(int k = 1; k <= K; k++) {
            for(int i = 1; i <= n; i++) {
                dp[i][k] = dp[i - 1][k]; //初始化i不在第k个矩阵
                for(int j = 0; j < i; j++) //枚举第k块起点j
                    dp[i][k] = max(dp[i][k], dp[j][k - 1] + s1[i] - s1[j]);
            }
        }
        cout << dp[n][K] << endl;
    }
    else { //两列的情况
        for(int i = 1; i <= n; i++)
            cin >> x >> y, s1[i] = s1[i - 1] + x, s2[i] = s2[i - 1] + y;
        for(int k = 1; k <= K; k++) {
            for(int i = 1; i <= n; i++) { //左边一列
                for(int j = 1; j <= n; j++) { //右边一列
                    f[i][j][k] = max(f[i - 1][j][k], f[i][j - 1][k]);
                    for(int l = 0; l < i; l++) //只有左边在第k个矩形
                        f[i][j][k] = max(f[i][j][k], f[l][j][k - 1] + s1[i] - s1[l]);
                    for(int l = 0; l < j; l++) //只有右边在第k个矩形
                        f[i][j][k] = max(f[i][j][k], f[i][l][k - 1] + s2[j] - s2[l]);
                    if(i == j) //两个都在第k个矩形
                        for(int l = 0; l < i; l++)
                            f[i][j][k] = max(f[i][j][k], f[l][l][k - 1] + s1[i] - s1[l] + s2[j] - s2[l]);
                }
            }
        }
        cout << f[n][n][K] << endl;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_40588429/article/details/84531559