BZOJ3544 [ONTAK2010]Creative Accounting 前缀和乱搞

维护模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;
}

你可能感兴趣的:(BZOJ3544 [ONTAK2010]Creative Accounting 前缀和乱搞)