解法一:暴力枚举
注意点:简单的三层循环顺序得到的不一定是和为最大值的情况
如:10,9,8,7,1
10+8+7 是大于 10+9+1 的
代码:
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
const int N = 100005;
long long n,mod;
long long a[N];
bool have_ans;
priority_queue<int>q;
int main()
{
cin>>n>>mod;
for(int i=1;i<=n;++i) cin>>a[i];
sort(a+1,a+n+1,greater<int>());
for(int i=1;i<=n;++i)
{
for(int j=i+1;j<=n;++j)
{
for(int k=j+1;k<=n;++k)
{
if((a[i]+a[j]+a[k])%mod==0)
{
q.push(a[i]+a[j]+a[k]);
}
}
}
}
cout<<q.top();
return 0;
}
解法二:dp优化
f[i][j] 表示 选取i个物品 余数为j的最大和数
f表示选取当前物品,g表示选取上一物品的结果
#include<iostream>
#include<cstring>
using namespace std;
const int N = 100005;
int n,mod;
int a;
int f[4][1005];
int g[4][1005];
int main()
{
cin>>n>>mod;
memset(f,-0X3f,sizeof f);
f[0][0]=0;
for(int i=1;i<=n;++i)
{
cin>>a;
memcpy(g,f,sizeof f);
for(int j=1;j<=3;j++)
{
for(int k=0;k<mod;k++)
{
//f[j][(a+k)%mod]=max(g[j][(a+k)%mod],g[j-1][k]+a);
f[j][k]=max(g[j][k],g[j-1][(k-(a%mod)+mod)%mod]+a);
}
}
}
cout<<f[3][0];
return 0;
}
优化3:只需选取余数为k的最大的三个数进行计算即可
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 100005;
int n,mod;
int a[N];
int f[4][1005];
int g[4][1005];
vector<int>v[1005];
int cnt;
int t;
int main()
{
cin>>n>>mod;
for(int i=1;i<=n;++i)
{
cin>>t;
v[t%mod].push_back(t);
}
for(int i=0;i<mod;i++)
{
sort(v[i].begin(),v[i].end(),greater<int>());
for(int j=0;j<3&&j<v[i].size();j++) a[++cnt]=v[i][j];
}
memset(f,-0X3f,sizeof f);
f[0][0]=0;
for(int i=1;i<=cnt;++i)
{
memcpy(g,f,sizeof f);
for(int j=1;j<=3;j++)
{
for(int k=0;k<mod;k++)
{
f[j][k]=max(g[j][k],g[j-1][(k-(a[i]%mod)+mod)%mod]+a[i]);
}
}
}
cout<<f[3][0];
return 0;
}