bzoj3544: [ONTAK2010]Creative Accounting（splay）

题记

bzoj3544泊兰题诶，不过是道权限题诶
所以我推荐你们..上波兰网站
不过这网站停用了【捂脸】多尴尬…

---

题意

给定一个长度为N的数组a和Mod，求一个区间[l,r]，使得$(\sum\limits_{i=l}^{r}{a_i}) mod M$的值最大，求出这个值.
注意这里的模是数学上的模，即负数时的情况

---

题解

这题吧，开始…脑残的想到了树状数组【捂脸】，然后数据范围10^18明显不可能（哭
维护前缀和，我们发现如果$a_i=3，a_j=2，[i+1,j]即跨越了一个模$，所以是很大的。那么我们便去寻找每一个点最多能得到多少，也就是找到它前面比它大的最小的，后面比它小的最大的，然后后面这一段会在它后面那个点往前找时被找到（也有可能找不到，但那时这一段也一定不是最优解），所以每次我们加入一个点时，就去找这个数的后继就好啦。结果就是max(a[i]+mod-后继，a[i])，然后我们发现，后继一定小于mod，所以a[i]+mod-后继>=a[i]（没有后继时等号成立），因此写的时候就..可以不比较a[i]+mod-后继和a[i]了

而且而且..这题…网上全是set…
而且而且，set还快 呜呜呜反正比我splay快呜呜呜
我可能写了个假的splay！
哼我要去学set（挖坑）

---

代码

#include <cstdio>
#include <algorithm>
using namespace std;
#define N 210000
int n,ch[N][2],fa[N],succ,pre,tot=0,root;
long long a[N],mod,v[N],mx=0;
void rotate(int &x){
    int y=fa[x],z=fa[y],t=ch[y][0]==x;
    if(z) ch[z][ch[z][1]==y]=x;
    else root=x;
    fa[x]=z;fa[y]=x;fa[ch[x][t]]=y;
    ch[y][t^1]=ch[x][t];ch[x][t]=y;
}
void splay(int x,int &rt){
    while(x!=rt){
        int y=fa[x];
        if(y!=rt){
            if(ch[y][0]==x^ch[fa[y]][0]==y) rotate(x);
            else rotate(y);
        }rotate(x);
    }
}
void query(int p,long long x){//求后继 
    if(v[p]>x){
        succ=p;
        if(ch[p][0]) query(ch[p][0],x);
    }else if(ch[p][1]) query(ch[p][1],x);
}
void insert1(int p,long long x){
    if(ch[p][1]) {insert1(ch[p][1],x);return;}
    v[++tot]=x;ch[tot][0]=ch[tot][1]=0;fa[tot]=p;ch[p][1]=tot; 
}
void insert(long long x){
    pre=ch[root][0];
    v[++tot]=x;ch[tot][0]=ch[tot][1]=0;
    if(succ){
        while(ch[pre][1]) pre=ch[pre][1];
        fa[tot]=pre;ch[pre][1]=tot;
    }else fa[tot]=root,ch[root][0]=tot;
}
int main(){
    scanf("%d%lld",&n,&mod);
    root=1;v[++tot]=ch[tot][0]=ch[tot][1]=fa[tot]=0;
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);a[i]%=mod;
        if(a[i]<0) a[i]+=mod;
        a[i]=(a[i-1]+a[i])%mod;
        succ=0;query(root,a[i]);
        mx=max(mx,(mod-v[succ]+a[i])%mod);
        if(succ) {splay(succ,root);insert(a[i]);}
        else insert1(root,a[i]);
    }printf("%I64d\n",mx);
    return 0;
}