【SCOI 2005】 最大子矩阵

【题目链接】

点击打开链接

【算法】

动态规划

我们发现，M只有两种取值，1和2，那么，只需分类讨论即可

当M = 1时，其实这个问题就成了就最大连续子段和的问题，只不过要选K段而已

用f[i][j]表示选到第i行，选了j段，那么，显然有 : f[i][j] = max{f[i-1][j],f[k][j-1] + sum[i] - sum[k]}（sum为前缀和）

当M = 2时，我们用f[i][j][k]表示第一列选到第i行，第二列选到第j行,选了k段

那么 ：

如果不取，f[i][j][k] = max{f[i-1][j][k],f[i][j-1][k]}

如果第一列取，f[i][j][k] = max{f[t][j][k-1] + sum[i][1] - sum[t][1]}

如果第二列取,   f[i][j][k] = max{f[i][t][k-1] + sum[i][2] - sum[t][2]}

如果第一二列都取，我们可以把它看成两种情况

第一种，看成一个大矩形，f[i][j][k] = max{f[t][t][k-1] + sum[i][1] - sum[t][1] + sum[i][2] - sum[t][2]}

第二种，看成两个小矩形，f[i][j][k] = max{f[t][t][k-2] + sum[i][1] - sum[t][1] + sum[i][2] - sum[t][2]} (k > 1)

此题就是需要我们认真读题，如果发现了"1<=M<=2"，那么，问题就变得简单多了！

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 110
#define MAXK 12
const int INF = 2e9;

int n,m,k;

inline void solve1()
{
        int i,j,t;
        static int a[MAXN],sum[MAXN],f[MAXN][MAXK];
        for (i = ; i <= n; i++)
        {
                for (j = ; j <= k; j++)
                {
                        f[i][j] = -INF;
                }
        }
        memset(sum,,sizeof(sum));
        for (i = ; i <= n; i++)
        {
                scanf("%d",&a[i]);
                sum[i] = sum[i-] + a[i];
        }
        for (i = ; i <= n; i++)
        {
                for (j = ; j <= k; j++)
                {
                        f[i][j] = f[i-][j];
                        for (t = ; t < i; t++)
                        {
                                f[i][j] = max(f[i][j],f[t][j-]+sum[i]-sum[t]);
                        }
                }
        }
        printf("%d\n",f[n][k]);
}
inline void solve2()
{
        int i,j,x,y;
        static int a[MAXN][],sum[MAXN][],f[MAXN][MAXN][MAXK];
        for (i = ; i <= n; i++)
        {
                for (j = ; j <= n; j++)
                {
                        for (x = ; x <= k; x++)
                        {
                                f[i][j][x] = -INF;
                        }
                }
        }
        memset(sum,,sizeof(sum));
        for (i = ; i <= n; i++)
        {
                scanf("%d%d",&a[i][],&a[i][]);
                sum[i][] = sum[i-][] + a[i][];
                sum[i][] = sum[i-][] + a[i][];
        }
        for (i = ; i <= n; i++)
        {
                for (j = ; j <= n; j++)
                {
                        for (x = ; x <= k; x++)
                        {
                                f[i][j][x] = max(f[i-][j][x],f[i][j-][x]);
                                for (y = ; y < i; y++) f[i][j][x] = max(f[i][j][x],f[y][j][x-]+sum[i][]-sum[y][]);
                                for (y = ; y < j; y++) f[i][j][x] = max(f[i][j][x],f[i][y][x-]+sum[j][]-sum[y][]);
                                if (i == j)
                                {
                                        for (y = ; y < i; y++)
                                        {
                                                f[i][j][x] = max(f[i][j][x],f[y][y][x-]+sum[i][]-sum[y][]+sum[i][]-sum[y][]);
                                                if (x > ) f[i][j][x] = max(f[i][j][x],f[y][y][x-]+sum[i][]-sum[y][]+sum[i][]-sum[y][]);
                                        }
                                }
                        }
                }
        }
        printf("%d\n",f[n][n][k]);
}

int main()
{

        scanf("%d%d%d",&n,&m,&k);
        if (m == ) solve1();
        else solve2();

        return ;
}