Jzoj1306 Sum

 给出一个数列a1，a2，。。。，an和K,P。 　　设Si,j = ai + ai+1::: + aj 　　Answer = min{Si,j mod P|Si,j mod P>=K}，其中i<=j，{Si,j mod P|Si,j mod P>=K}非空。

先预处理出取模后的前缀和s，那么问题变成min(s[j]-s[i]%P)且s[j]-s[i]%P>=k

把s序列反过来作，对于每个数s[i],维护一个set存有s[i]+1~s[n],让后分两种情况讨论

1.s[j]>=s[i]，这种情况可以在set中寻找比s[i]+k大的最小的数

2.s[j]<s[i]，这种情况可以在set中寻找比s[i]+k-p大的最小的数，并且判断这个数是否<s[i]

最后统计answer

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<set>
using namespace std;
int n,k,p,s[100010],ans=1<<29;
set<int> w;
int main(){
	scanf("%d%d%d",&n,&k,&p);
	for(int i=1;i<=n;++i){
		scanf("%d",s+i);
		(s[i]+=s[i-1])%=p;
	}
	w.insert(s[n]);
	set<int>::iterator it;
	for(int i=--n;i;--i){
		it=w.lower_bound(s[i]+k);
		if(it!=w.end()) ans=min(ans,*it-s[i]);
		it=w.lower_bound(k-p+s[i]);
		if(it!=w.end() && *it<s[i]) ans=min(ans,p+*it-s[i]);
		w.insert(s[i]);
	}
	printf("%d
",ans);
}