[ARC082F] Sandglass（线段树）

Description

有一个沙漏由两个上下相通玻璃球 $A$ 和 $B$ 构成，这两个玻璃球都含有一定量的沙子，我们暂且假定 $AB$ 中位于上方的玻璃球的为 $U$ ，下方的玻璃球为 $L$ ，则除非 $U$ 中没有沙子，否则每秒钟都会有1克沙子从 $U$ 掉入 $L$ 。

在第 $0$ 个时刻， $A$ 中有 $a$ 克沙子， $B$ 中有 $X−a$ 克沙子(总共有 $X$ 克沙子)，且 $U$ 为 $A$ ， $L$ 为 $B$ (即 $A$ 上 $B$ 下)。

在 $r1,r2,...,rK$ 这些时刻，我们将倒转整个沙漏，使得原来的 $U$ 变成 $L$ ，原来的 $L$ 变成 $U$ 。对于翻转操作，t时刻是指从第 $0$ 个时刻起经过 $t$ 秒后的时刻，我们可以将翻转沙漏的操作看做瞬间完成的。

现在有 $Q$ 次询问，每一次询问会给定一对非负整数（ $t_i$ ， $a_i$ ），求 $a=a_i$ ，第 $t_i$ 时刻， $A$ 中所含沙子的克数。

Input

第一行一个正整数X

第二行一个正整数K

第三行K个整数，表示r1,r2,…,rK

接下来一行一个正整数Q

接下来Q行，每行两个非负整数，分别表示每次次询问的(ti,ai)

Output

一共Q行

对于每次询问，输出一行一个非负整数表示答案。

题解：

神题啊！

这题可以用线段树做，我们维护 1~X 中，每个 a 经过一系列操作后的值，我们只用求每个询问中的 a，为了避免爆空间，我们动态开点。

我们把询问按时间排序，同时操作 r 数组，如果A再上面就减区间长度，和0取max，不然就加，与X取min。

像这样：

for(int i=0,j=1,rev=0;i<=n;i++,rev^=1){
    while(j<=m&&q[j].t<=r[i+1]){
        int tmp=query(root,1,X+1,q[j].a);
        if(rev)tmp=min(X,tmp+q[j].t-r[i]);
        else tmp=max(0,tmp-q[j].t+r[i]);
        ans[q[j++].id]=tmp;
    }
    if(j>m)break;
    int tmp=r[i+1]-r[i];
    if(rev)Add(root,1,X+1,tmp);
    else Sub(root,1,X+1,tmp);
}

然后每个线段树节点维护当前区间的最小值和最大值。

加的话，这样：

if(minv[o]>=X-x)
    maxv[o]=minv[o]=set[o]=X,add[o]=0;
else if(maxv[o]<=X-x)
    maxv[o]+=x,minv[o]+=x,add[o]+=x;

减就这样：

if(maxv[o]<=x)
    maxv[o]=minv[o]=add[o]=set[o]=0;
else if(minv[o]>=x)
    maxv[o]-=x,minv[o]-=x,add[o]-=x;

思路还是很简单的，这题主要还是码码。

COMPLETE CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int n,m,X,tot=0,root,r[100005];
int lch[8000005],rch[8000005];
int minv[8000005],maxv[8000005];
int set[8000005],add[8000005];
struct Question{
    int t,a,id;
}q[100005];
int ans[100005];

bool comp(Question a,Question b){return a.t<b.t;}

int newnode(int &o,int x,int y){
    o=++tot,minv[o]=x,maxv[o]=y,set[o]=-1;
}

void pushup(int o){
    minv[o]=min(minv[lch[o]],minv[rch[o]]);
    maxv[o]=max(maxv[lch[o]],maxv[rch[o]]);
}

void pushdown(int o,int l,int r){
    int mid=l+r>>1;
    int &lc=lch[o],&rc=rch[o];
    if(!lc)newnode(lc,l,mid-1);
    if(!rc)newnode(rc,mid,r-1);
    if(~set[o]){
        minv[lc]=maxv[lc]=set[lc]=set[o];
        minv[rc]=maxv[rc]=set[rc]=set[o];
        add[lc]=add[rc]=0,set[o]=-1;
    }
    if(add[o]){
        minv[lc]+=add[o],maxv[lc]+=add[o],add[lc]+=add[o];
        minv[rc]+=add[o],maxv[rc]+=add[o],add[rc]+=add[o];
        add[o]=0;
    }
}

void Sub(int &o,int l,int r,int x){
    if(o==0)newnode(o,l,r-1);
    if(maxv[o]<=x)
        maxv[o]=minv[o]=add[o]=set[o]=0;
    else if(minv[o]>=x)
        maxv[o]-=x,minv[o]-=x,add[o]-=x;
    else{
        pushdown(o,l,r);
        int mid=l+r>>1;
        Sub(lch[o],l,mid,x),Sub(rch[o],mid,r,x);
        pushup(o);
    }
}

void Add(int &o,int l,int r,int x){
    if(o==0)newnode(o,l,r-1);
    if(minv[o]>=X-x)
        maxv[o]=minv[o]=set[o]=X,add[o]=0;
    else if(maxv[o]<=X-x)
        maxv[o]+=x,minv[o]+=x,add[o]+=x;
    else{
        pushdown(o,l,r);
        int mid=l+r>>1;
        Add(lch[o],l,mid,x),Add(rch[o],mid,r,x);
        pushup(o);
    }
}

int query(int o,int l,int r,int x){
    if(o==0)newnode(o,l,r-1);
    if(r-l==1)return minv[o];
    pushdown(o,l,r);
    int mid=l+r>>1;
    if(x<mid)return query(lch[o],l,mid,x);
    else return query(rch[o],mid,r,x);
}

int main(){
    scanf("%d%d",&X,&n);
    for(int i=1;i<=n;i++)scanf("%d",r+i);
    r[n+1]=1e9;
    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,comp);
    for(int i=0,j=1,rev=0;i<=n;i++,rev^=1){
        while(j<=m&&q[j].t<=r[i+1]){
            int tmp=query(root,1,X+1,q[j].a);
            if(rev)tmp=min(X,tmp+q[j].t-r[i]);
            else tmp=max(0,tmp-q[j].t+r[i]);
            ans[q[j++].id]=tmp;
        }
        if(j>m)break;
        int tmp=r[i+1]-r[i];
        if(rev)Add(root,1,X+1,tmp);
        else Sub(root,1,X+1,tmp);
    }
    for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
}