CF1433F Zero Remainder Sum——dp

一看题就直到是老dp了。
给出一个矩形,每行最多只能选一半向下取整,问最后结果最大并且整除k的值是多少。
显然
我们要构造一个四维数组
dp[i][j][k][h]
代表在i行j列选取k个并且取模为h的结果
code:

#include<bits/stdc++.h>
using namespace std;
int a[100][100];
int dp[75][75][75][75];
int main()
{
    
    
	int n,m,k;
	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(dp,-1,sizeof dp);
	dp[1][0][0][0] = 0;
	for(int i=1;i<=n;i++)
	{
    
    
		for(int j=1;j<=m;j++)
		{
    
    
			for(int kk=0;kk<=min(m/2,j);kk++)
			{
    
    
				for(int h=0;h<k;h++)
				{
    
    
					if(kk!=0&&dp[i][j-1][kk-1][h]!=-1)
					dp[i][j][kk][(h+a[i][j])%k] = max(dp[i][j-1][kk-1][h]+a[i][j],dp[i][j][kk][(h+a[i][j])%k]);
					dp[i][j][kk][h] = max(dp[i][j-1][kk][h],dp[i][j][kk][h]);
				}
				for(int h=0;h<k;h++)
				{
    
    
					dp[i+1][0][0][h] = max(dp[i+1][0][0][h],dp[i][m][kk][h]);
				}
			}
			
		}
	}
	int ans = 0;
	for(int i=0;i<=m/2;i++)
	{
    
    
		ans = max(ans,dp[n][m][i][0]);
	}
	cout<<ans<<endl;
 }

猜你喜欢

转载自blog.csdn.net/naiue/article/details/109366845