bzoj 3544 [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

题解

 

题解：因为，mod的最大值是其后者，或者第一个，

然后贪心即可，set维护，n log n

#include<cstring>
#include<cmath>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<set>

#define N 200007
#define ll long long
using namespace std;
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}

ll n,m,ans;
ll a[N],sum[N];
set<ll>q;

int main()
{
    n=read(),m=read();
    for (int i=1;i<=n;i++)a[i]=(read()%m+m)%m;
    for (int i=1;i<=n;i++)(sum[i]=(sum[i-1]+a[i])%m+m)%=m;
    q.insert(0);
    for (int i=1;i<=n;i++)
    {
        ll t;
        if (q.upper_bound(sum[i])==q.end()) t=*q.begin();
        else t=*q.upper_bound(sum[i]);
        ans=max(ans,((sum[i]-t)%m+m)%m);
        q.insert(sum[i]);
    }
    printf("%lld",ans);
}