51Nod1472 取余最大值
题目看这里
又是一个七级题目
妥妥的分治啊,枚举右端点,左边分两段计数就好啦~
关于标解:
网上好像很多都是前缀和+二分之类的,反正应该差不多快吧
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 300010
#define LL long long
using namespace std;
LL S=0;
int v[N],s[N],c[N*4],c2[N*4],mx[N],n,M,wx[N];
inline void cdq(int l,int r){
if(l==r){ return; }
int m=l+r>>1; cdq(l,m); cdq(m+1,r);
s[l-1]=0;
for(int i=l;i<=m;++i) s[i]=(s[i-1]+v[i])%M;
s[m+1]=v[m+1]%M; mx[m+1]=v[m+1];
for(int i=m+2;i<=r;++i){
s[i]=(s[i-1]+v[i])%M;
mx[i]=max(mx[i-1],v[i]);
}
mx[m]=v[m];
++c2[((s[m]-s[m-1]-mx[m])%M+M)%M];
for(int i=m-1;i>=l;--i){
mx[i]=max(mx[i+1],v[i]);
++c2[((s[m]-s[i-1]-mx[i])%M+M)%M];
}
for(int i=m,j=m+1;j<=r;++j){
while(i>=l && mx[i]<mx[j]){
++c[(s[m]-s[i-1]+M)%M];
--c2[((s[m]-s[i-1]-mx[i])%M+M)%M];
--i;
}
S+=c[(M-s[j]+mx[j]+M)%M];
S+=c2[(M-s[j]+M)%M];
}
for(int i=l;i<=m;++i) c[(s[m]-s[i-1]+M)%M]=c2[((s[m]-s[i-1]-mx[i])%M+M)%M]=0;
}
int main(){
scanf("%d%d",&n,&M);
for(int i=1;i<=n;++i) scanf("%d",v+i);
cdq(1,n); printf("%lld\n",S);
}