dp WA求条
查看原帖
dp WA求条
877829
Anythingispossible楼主2024/8/21 15:32

设 dpi,jdp_{i,j} 表示以 ii 为第 jj 个数能满足条件的最大和。

状态转移方程: dpi,j=max⁡(max⁡k=1i−1dpk,j−1+ai,dpi,j((dpk,j−1+ai)mod  d=0))dp_{i,j}=\max(\max_{k=1}^{i-1} dp_{k,j-1}+a_i,dp_{i,j}((dp_{k,j-1}+a_i) \mod d =0))

代码:

#include<cstdio>
#include<iostream>
using namespace std;
const int N=105;
long long n,m,d,a[N],dp[N][N],ans=-1e15;
int main(){
	scanf("%lld %lld %lld",&n,&m,&d);
	if(m>n){
		printf("-1");
		return 0;
	}
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
	} 
	for(int i=1;i<=n;i++){
		if(a[i]%d==0) dp[i][1]=a[i];
		else dp[i][1]=-1e15;
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			for(int k=1;k<i;k++){
				if((dp[k][j-1]+a[i])%d==0) dp[i][j]=max(dp[i][j],dp[k][j-1]+a[i]);
			}
		}
	}
	for(int i=m;i<=n;i++) ans=max(ans,dp[i][m]);
	if(ans<0) printf("-1");
	else printf("%lld",ans);
	return 0;
}
2024/8/21 15:32
加载中...