【DP+分组背包】 Zero Remainder Sum
题面:Codeforces Round #677 (Div. 3) F. Zero Remainder Sum
题意:给出一个n×m的数字矩阵,和一个k (1 ≤ n,m,k ≤70)。然后可以在每行选出不超过m/2个数(也可以不选),使得最后总和最大且是k的倍数。
思路:
首先根据题意,可以知道取数的方法与分组背包类似,分组背包不会的话评论抠1,我再补充。我们可以将一行看作一组,每组最多可以取m/2个,如果没有k的限制,那么这题以及结束了,分组背包套一下就行了。但是考虑到有k的限制,我们需要对每个状态记一个整除 k 的余数,但是这样就不好dp了,又因为n,m,k的范围很小,所以我们可以多开两维数组,来保存当前状态的取过的数的个数 和 当前的余数。
最后再借鉴分组背包的转移,我们就可以得到一个n^4的dp了:
dp [i] [j] [cnt] [res] 表示在第i行第j列,取过cnt个数,余数为res的最大值
转移可以看代码,按照第j+1个取或者不取来转移
几个细节提醒一下:
1.dp要初始化
2.转移顺序要注意,一开始我考虑顺推,但是顺推是要根据余数转移的,不好写,写炸了,后来调代码也乱调,最后在257组样例wa了,就放弃顺推了。考虑从后往前推,这样的话只要考虑第j+1个就行
3.转移到新的一行时,要将这行的初值赋为上一行的最大值
4.输出时要和0取max,因为可能不能组成任何一个 k 的倍数,这时候我们应该每行都不取,最终答案是0
#include<bits/stdc++.h>
#define ll long long
#define mem(b,t) memset(b,t,sizeof(b))
#define INF 0x3f3f3f3f
#define rep(i,n,m) for(int i=n;i<=m;i++)
#define per(i,n,m) for(int i=n;i>=m;i--)
#define bug1(i) cout<<i<<endl
#define bug2(i,j) cout<<i<<" "<<j<<endl
#define mod 1000000007
#define maxn 200005
#define eps 1e-6
#define pi acos(-1)
using namespace std;
int dp[75][75][75][75];
int a[75][75];
int main(){
mem(dp,-1);//初始化为-1 代表取不到
int n,m,k;
cin>>n>>m>>k;
rep(i,1,n) rep(j,1,m) cin>>a[i][j];
dp[1][0][0][0]=0;
rep(i,1,n){
rep(j,0,m-1){
rep(cnt,0,m/2){
rep(res,0,k-1){//背包转移
dp[i][j+1][cnt][res]=max(dp[i][j+1][cnt][res],dp[i][j][cnt][res]);
if(cnt&&dp[i][j][cnt-1][res]!=-1)
dp[i][j+1][cnt][(res+a[i][j+1])%k]=max(dp[i][j+1][cnt][(res+a[i][j+1])%k],dp[i][j][cnt-1][res]+a[i][j+1]);
}
rep(res,0,k-1){//更新下一行开始的值
dp[i+1][0][0][res]=max(dp[i+1][0][0][res],dp[i][j+1][cnt][res]);
}
}
}
}
cout<<max(0,dp[n+1][0][0][0])<<endl;
return 0;
}
/*
4 2 49
16 42
58 37
2 17
40 61
*/