暑假集训Day6 K(线段树模型)

题目链接在这里：Problem - K - Codeforces

经过观察可以发现会见骑士结束的时间点可以表示成一个式子c_x=max(t_i+sigma(d_i+...d_x)) (i=1...x)

只需要把 t_i 离散化出来，这就是跟上一个一样的经典线段树模型。

码力要加强啊！写的时间太长了！

#include "bits/stdc++.h"
#define lson rt<<1,l,m
#define rson rt<<1|1,m+1,r
using namespace std;
typedef long long LL;
const int MAX=1e6+5;
LL t,n,a[MAX],b[MAX],ss[MAX];
LL cc[MAX],tr[MAX<<2],la[MAX<<2];
struct Que{
    char c;
    int x,y;
}que[MAX];
void update(LL x,LL y,LL nn){for (;x<=nn;x+=(x&-x)) cc[x]+=y;}
LL find(LL x){LL an=0;for (;x>0;x-=(x&-x)) an+=cc[x];return an;}
void PushDown(LL rt){
    if (la[rt]){
        la[rt<<1]+=la[rt];
        la[rt<<1|1]+=la[rt];
        tr[rt<<1]+=la[rt];
        tr[rt<<1|1]+=la[rt];
        la[rt]=0;
    }
}
void PushUp(LL rt){tr[rt]=max(tr[rt<<1],tr[rt<<1|1]);}
void upd(LL rt,LL l,LL r,LL x,LL y,LL z){
    if (x<=l && r<=y){
        la[rt]+=z;
        tr[rt]+=z;
        return;
    }
    LL m=(l+r)>>1;
    PushDown(rt);
    if (x<=m) upd(lson,x,y,z);
    if (y>m) upd(rson,x,y,z);
    PushUp(rt);
}
LL search(LL rt,LL l,LL r,LL x,LL y){
    if (x<=l && r<=y) return tr[rt];
    LL m=(l+r)>>1,an=-1;
    PushDown(rt);
    if (x<=m) an=max(an,search(lson,x,y));
    if (y>m) an=max(an,search(rson,x,y));
    return an;
}
int main(){
    freopen ("k.in","r",stdin);
    freopen ("k.out","w",stdout);
    LL i,j,n=0,pos,zt;
    memset(cc,0,sizeof(cc));
    memset(tr,0,sizeof(tr));
    memset(la,0,sizeof(la));
    scanf("%lld",&t);
    for (i=1;i<=t;i++){
        scanf("\n%c",&que[i].c);
        if (que[i].c=='?' || que[i].c=='-') scanf("%lld",&que[i].x);
        else{
            scanf("%lld%lld",&que[i].x,&que[i].y);
            ss[++n]=que[i].x;
        }
    }
    sort(ss+1,ss+n+1);
    ss[n+1]=1e15;
    for (i=1;i<=t;i++){
        if (que[i].c=='?'){
            pos=lower_bound(ss+1,ss+n+1,que[i].x)-ss;
            if (ss[pos]>que[i].x) pos--;
            if (pos>=1) zt=search(1,1,n,1,pos);
            else zt=0;
            printf("%lld\n",max(0ll,find(pos)+zt-que[i].x));
        }
        else if (que[i].c=='+'){
            pos=lower_bound(ss+1,ss+n+1,que[i].x)-ss;
            update(pos,que[i].y,n);
            upd(1,1,n,pos,pos,que[i].x);
            if (pos!=n)upd(1,1,n,pos+1,n,-que[i].y);
        }
        else if (que[i].c=='-'){
            pos=lower_bound(ss+1,ss+n+1,que[que[i].x].x)-ss;
            update(pos,-que[que[i].x].y,n);
            upd(1,1,n,pos,pos,-que[que[i].x].x);
            if (pos!=n) upd(1,1,n,pos+1,n,que[que[i].x].y);
        }
    }
    return 0;
}