BZOJ1084 SCOI2005最大子矩阵

考虑DP

f[i][j][k]表示一行到i一行到j共取k块最大值，类似于最长公共子序列n^2那种

注意相等时可以一起拿

By：大奕哥

#include<bits/stdc++.h>
using namespace std;
int n,m,K,g[105][105][15],f[105][105],s[105][105],x;
int main()
{
    scanf("%d%d%d",&n,&m,&K);
    for(int i=1;i<=n;++i)
    for(int j=1;j<=m;++j)
    {
        scanf("%d",&x);
        s[i][j]=s[i-1][j]+x;
    }
    if(m==1)
    {
        memset(f,-0x3f,sizeof(f));
        for(int i=1;i<=n;++i)f[i][0]=0; 
        for(int i=1;i<=n;++i)
        {
            for(int k=1;k<=K;++k)
            {
                f[i][k]=max(f[i][k],f[i-1][k]);
                for(int j=0;j<i;++j)
                f[i][k]=max(f[i][k],f[j][k-1]+s[i][1]-s[j][1]);
            }
        }
        printf("%d
",f[n][K]);
    }
    else
    {
        memset(g,-0x3f,sizeof(g));
        for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
        g[i][j][0]=0;
        for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
        {
            for(int k=1;k<=K;++k)
            {
                g[i][j][k]=max(g[i-1][j][k],g[i][j-1][k]);
                for(int p=0;p<i;++p)g[i][j][k]=max(g[i][j][k],g[p][j][k-1]+s[i][1]-s[p][1]);
                for(int p=0;p<j;++p)g[i][j][k]=max(g[i][j][k],g[i][p][k-1]+s[j][2]-s[p][2]);
                if(i==j)
                for(int p=0;p<i;++p)
                g[i][j][k]=max(g[i][j][k],g[p][p][k-1]+s[i][1]+s[i][2]-s[p][1]-s[p][2]);
            }
        }
        printf("%d
",g[n][n][K]);
    }
    return 0;
}