维护模m意义下的前缀和,然后从头到尾扫一遍,把扫到的数字加入set里,然后在set里查找upper_bound,这样保证肯定能拿到最大的答案。
具体见代码。
//BZOJ3544
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<vector>
#include<set>
#include<cstdlib>
using namespace std;
typedef long long LL;
const int MAXN=200020;
int n;
LL a[MAXN],s[MAXN],ans,p,m;
set<LL> S;
int main()
{
scanf("%d%lld",&n,&m);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int i=1;i<=n;i++)
{
if(a[i]<0) a[i]=a[i]+(abs(a[i]/m)+1)*m;
a[i]%=m;
}
for(int i=1;i<=n;i++) s[i]=(s[i-1]+a[i])%m;
for(int i=1;i<=n;i++)
{
ans=max(ans,s[i]);
if(S.upper_bound(s[i])!=S.end())
{
p=*(S.upper_bound(s[i]));
ans=max(ans,(s[i]+m-p)%m);
}
S.insert(s[i]);
}
printf("%lld\n",ans);
return 0;
}