【BZOJ3544】[ONTAK2010]Creative Accounting 前缀和+set

【BZOJ3544】[ONTAK2010]Creative Accounting

Description

给定一个长度为N的数组a和M,求一个区间[l,r],使得(\sum_{i=l}^{r}{a_i}) mod M的值最大,求出这个值,注意这里的mod是数学上的mod

Input

第一行两个整数N,M。
第二行N个整数a_i。

Output

输出一行,表示答案。

Sample Input

5 13
10 9 5 -5 7

Sample Output

11

HINT

【数据范围】
N<=200000,M,a_i<=10^18

题解:首先子串和=两个前缀相减。所以对于每个前缀和,我们找到max(它-之前的前缀)就行了。贪心可知之前的前缀和要么是它的后继要么是0。

#include <cstdio>
#include <cstring>
#include <set>
#include <iostream>
using namespace std;
typedef long long ll; set<ll> s;
set<ll>::iterator it;
ll m,sum,ans;
int n;
inline ll rd()
{
ll ret=0,f=1; char gc=getchar();
while(gc<'0'||gc>'9') {if(gc=='-')f=-f; gc=getchar();}
while(gc>='0'&&gc<='9') ret=ret*10+gc-'0',gc=getchar();
return ret*f;
}
int main()
{
n=rd(),m=rd();
int i;
s.insert(0);
for(i=1;i<=n;i++)
{
sum=((sum+rd())%m+m)%m;
it=s.upper_bound(sum);
if(it!=s.end()) ans=max(ans,sum+m-(*it));
s.insert(sum);
ans=max(ans,sum);
}
printf("%lld",ans);
return 0;
}
上一篇:Beta版本——第四次冲刺博客


下一篇:Zero Sum Subarray