D - Max Multiple ABC281-CSDN博客

本文链接：https://blog.csdn.net/m0_63305704/article/details/128272697

D - Max Multiple (atcoder.jp)

题意是给你n个数，让你在里面选k个，问你最大的d的倍数是多少

分析:

昨天赛时我想到了背包，想到了dp。但是因为缺乏经验，也没敢写...

其实就是个很典的DP

借用一下背包的思想

把集合分为第i个数选还是不选

令 $f[i][j][k]$ 为前i个中选j个取模为k的最大值

那么最后的答案就是 $f[n][k][0]$

状态转移：

不选第i个： $f[i][j][k]=f[i-1][j][k];$

选第i个： $f[i][j][k]=f[i-1][j-1][((k-a[i])%d+d)%d]+a[i];$

这里有一个减法取模的一个应用，在+d之后再模一下。这也是为什么官方题解上用的是a[i+1]，那就是避免了减法取模

所以每次状态转移的过程中取两者的最大值即可

再注意一下边界细节之类的

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#define IOS ios::sync_with_stdio(false), cin.tie(0);
#include<iostream>
#include<map>
#include<set> 
#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
#include<algorithm>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PAII;
const int N=2e6+10,M=5050,INF=1e18,mod=998244353;
int f[110][110][110],a[N];
signed main(){
    //IOS;
    int T;
    T=1;
    //cin>>T;
    while(T--)
    {
		int n,k,d;
		cin>>n>>k>>d;
		for(int i=1;i<=n;i++) cin>>a[i];
		memset(f,-0x3f,sizeof(f));
		for(int i=0;i<=n;i++) f[i][0][0]=0;
		for(int i=1;i<=n;i++)
			for(int j=1;j<=min(i,k);j++)
				for(int z=0;z<d;z++)
					f[i][j][z]=max(f[i-1][j][z],f[i-1][j-1][((z-a[i])%d+d)%d]+a[i]);
		if(f[n][k][0]<0) cout<<"-1";
		else cout<<f[n][k][0];
	}
    return 0;
} 
/*
令f[i][j][k]为前i个数选择了j个取模为k的最大值
有一种状态机的思想
每个数是选还是不选
不选：f[i][j][k]=f[i-1][j][k];
选:f[i][j][k]=f[i-1][j-1][((k-a[i])%d+d)%d+a[i]]; 
两者取最大值 

*/