CF1433F Zero Remainder Sum(四维DP+滚动数组优化)

题意:

给出一个二维数组,和三个数N M K,请你在每一行选少于等于M/2个数字,所有行选的数字的和在被K整除的情况下的最大值。

题解:

/*
f(i,j,l,r)表示在第i行第j位,选了l个数字,且此时模数为r的最大和 
 */ 
#include<bits/stdc++.h>
using namespace std;
const int maxn=80;
typedef long long ll;
long long f[2][maxn][maxn][maxn];
int a[maxn][maxn];
int n,m,k;
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",&a[i][j]);
    memset(f,-1,sizeof(f));
    f[1][0][0][0]=0;
    int F=1;
    for (int i=1;i<=n;i++,F^=1) for (int j=0;j<m;j++) for (int l=0;l<=min(j+1,m/2);l++) {
         for (int r=0;r<k;r++) {
             f[F][j+1][l][r]=max(f[F][j+1][l][r],f[F][j][l][r]);
                 if (l&&f[F][j][l-1][r]!=-1) {
                     f[F][j+1][l][(r+a[i][j+1])%k]=max(f[F][j+1][l][(r+a[i][j+1])%k],f[F][j][l-1][r]+a[i][j+1]);
                }
        }
        for (int r=0;r<k;r++) f[F^1][0][0][r]=max(f[F^1][0][0][r],f[F][j+1][l][r]);
    }
    printf("%lld\n",max(0ll,f[F][0][0][0]));
}

 

posted @ 2020-10-23 16:31  zlc0405  阅读(188)  评论(1编辑  收藏  举报