1. 程式人生 > 實用技巧 >CF1433F Zero Remainder Sum(四維DP+滾動陣列優化)

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])); }