AcWing 1234. 倍数问题
原题链接
中等
作者:
Nazarena
,
2021-03-01 21:05:16
,
所有人可见
,
阅读 445
C++ 代码
#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
int n,m;
vector<int> a[N];
int f[4][N]; // 背包问题空间优化
int main()
{
scanf("%d%d",&n, &m);
for(int i=0;i<n;i++)
{
int x;
scanf("%d",&x);
a[x % m].push_back(x);
}
memset(f,-0x3f,sizeof f);
f[0][0]=0;
for(int i=0;i<m;i++)
{
sort(a[i].begin(),a[i].end());
reverse(a[i].begin(),a[i].end());
for(int u=0;u<3 && a[i].size(); u ++ )
{
int x=a[i][u];
for(int j=3;j>=1;j--)
for(int k=0;k<m;k++)
f[j][k] = max(f[j][k], f[j-1][(k - x % m + m) % m] + x);
}
}
printf("%d\n",f[3][0]);
return 0;
}