【ARC082F】Sandglass【线段树】

我们可以首先把所有询问按时间排序，然后维护一棵以初始沙子数量为下标的线段树。把反转操作依次进行，相当于区间加减，然后区间对0取max，对x取min。这些可以简单地通过线段树实现。在执行操作的过程中询问答案。但是下标有$10^9$的范围，怎么办？我们动态开点就好。

#include<bits/stdc++.h>
using namespace std;
const int N=100005;
int x,n,m,root,tot,r[N],ans[N];
struct query{
    int t,a,id;
}q[N];
bool cmp(query a,query b){
    return a.t<b.t;
}
struct node{
    int lc,rc,tag,add,maxn,minn;
}t[N*50];
int newnode(int l,int r){
    tot++;
    t[tot].lc=t[tot].rc=t[tot].add=0;
    t[tot].tag=-1;
    t[tot].maxn=r;
    t[tot].minn=l;
    return tot;
}
void pushdown(int o,int l,int r){
    int mid=(l+r)/2;
    if(!t[o].lc){
        t[o].lc=newnode(l,mid);
    }
    if(!t[o].rc){
        t[o].rc=newnode(mid+1,r);
    }
    if(t[o].tag!=-1){
        t[t[o].lc].maxn=t[t[o].rc].maxn=t[t[o].lc].minn=t[t[o].rc].minn=t[t[o].lc].tag=t[t[o].rc].tag=t[o].tag;
        t[t[o].lc].add=t[t[o].rc].add=0;
        t[o].tag=-1;
    }
    if(t[o].add){
        t[t[o].lc].add+=t[o].add;
        t[t[o].rc].add+=t[o].add;
        t[t[o].lc].minn+=t[o].add;
        t[t[o].rc].minn+=t[o].add;
        t[t[o].lc].maxn+=t[o].add;
        t[t[o].rc].maxn+=t[o].add;
        t[o].add=0;
    }
}
void pushup(int o,int l,int r){
    int mid=(l+r)/2;
    if(!t[o].lc){
        t[o].lc=newnode(l,mid);
    }
    if(!t[o].rc){
        t[o].rc=newnode(mid+1,r);
    }
    t[o].minn=min(t[t[o].lc].minn,t[t[o].rc].minn);
    t[o].maxn=max(t[t[o].lc].maxn,t[t[o].rc].maxn);
}
void update1(int &o,int l,int r,int v){
    if(!o){
        o=newnode(l,r);
    }
    pushdown(o,l,r);
    if(t[o].minn+v>=x){
        t[o].maxn=t[o].minn=t[o].tag=x;
        t[o].add=0;
        return;
    }else if(t[o].maxn+v<=x){
        t[o].maxn+=v;
        t[o].minn+=v;
        t[o].add+=v;
        return;
    }
    int mid=(l+r)/2;
    update1(t[o].lc,l,mid,v);
    update1(t[o].rc,mid+1,r,v);
    pushup(o,l,r);
}
void update2(int &o,int l,int r,int v){
    if(!o){
        o=newnode(l,r);
    }
    pushdown(o,l,r);
    if(t[o].maxn<=v){
        t[o].maxn=t[o].minn=t[o].tag=0;
        t[o].add=0;
        return;
    }else if(t[o].minn>=v){
        t[o].maxn-=v;
        t[o].minn-=v;
        t[o].add-=v;
        return;
    }
    int mid=(l+r)/2;
    update2(t[o].lc,l,mid,v);
    update2(t[o].rc,mid+1,r,v);
    pushup(o,l,r);
}
int query(int &o,int l,int r,int k){
    if(!o){
        o=newnode(l,r);
    }
    if(l==r){
        return t[o].minn;
    }
    pushdown(o,l,r);
    int mid=(l+r)/2;
    if(k<=mid){
        return query(t[o].lc,l,mid,k);
    }else{
        return query(t[o].rc,mid+1,r,k);
    }
}
int main(){
    scanf("%d%d",&x,&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&r[i]);
    }
    r[n+1]=2e9;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].t,&q[i].a);
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    for(int i=0,j=1;i<=n;i++){
        while(j<=m&&q[j].t<=r[i+1]){
            int res=query(root,0,x,q[j].a);
            if(i&1){
                res=min(res+(q[j].t-r[i]),x);
            }else{
                res=max(res-(q[j].t-r[i]),0);
            }
            ans[q[j++].id]=res;
        }
        if(j>m){
            break;
        }
        if(i&1){
            update1(root,0,x,r[i+1]-r[i]);
        }else{
            update2(root,0,x,r[i+1]-r[i]);
        }
    }
    for(int i=1;i<=m;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}