bzoj1084: [SCOI2005]最大子矩陣(Dp)

Hanks_o發表於2018-03-01

題目傳送門

解法:
有點噁心啊。
一開始狀態表示錯想半天想不出。
%了下題解發現原來這樣表示狀態就可以了。
一列的很好解決。
兩列的就s[i][j][k]表示第一列取到第i個,第二列取到第j個,一共k個子矩陣的最大分值。
然後四種狀態轉移。
這個位置可以不選:s[i][j][t]=max(s[i-1][j][t],s[i][j-1][t]);

然後第一列和第二列分別往前移一段。
很好轉移。

如果i==j的話表示選到同一個位置了那麼我們可以一起往前弄矩陣。
詳情看程式碼。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
int a[3][110],sum[3][110],f[110][15];
int s[110][110][15];
int main() {
    int n,m,K;scanf("%d%d%d",&n,&m,&K);
    for(int j=1;j<=n;j++)for(int i=1;i<=m;i++)scanf("%d",&a[i][j]);swap(n,m);
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)sum[i][j]=sum[i][j-1]+a[i][j];
    if(n==1) {
        for(int j=1;j<=m;j++)for(int t=1;t<=K;t++) {
            f[j][t]=f[j-1][t];
            for(int k=0;k<j;k++)f[j][t]=max(f[j][t],f[k][t-1]+sum[1][j]-sum[1][k]);
        }printf("%d\n",f[m][K]);
    }else {
        for(int i=1;i<=m;i++)for(int j=1;j<=m;j++)for(int t=1;t<=K;t++) {
            s[i][j][t]=max(s[i-1][j][t],s[i][j-1][t]);
            for(int k=0;k<i;k++)s[i][j][t]=max(s[i][j][t],s[k][j][t-1]+sum[1][i]-sum[1][k]);
            for(int k=0;k<j;k++)s[i][j][t]=max(s[i][j][t],s[i][k][t-1]+sum[2][j]-sum[2][k]);
            if(i==j)for(int k=0;k<i;k++)s[i][j][t]=max(s[i][j][t],s[k][k][t-1]+sum[1][i]-sum[1][k]+sum[2][j]-sum[2][k]);
        }printf("%d\n",s[m][m][K]);
    }
    return 0;
}

相關文章