【国集作业】【ARC082F】Sandglass【线段树】

线段树好题，然而我并没有想到线段树
小视野链接
洛谷链接

题目大意

有一个沙漏，共$X$克沙子，每一秒就有一克沙子从上方掉到下方（掉完为止）。在第$R_1,R_2\ldots R_k$秒沙漏会被翻过来（即上下方的沙子质量交换）（翻转不计时）。有$Q$组询问：每次给出$\left( x_i,t_i\right)$，求初始上方有$x_i$克沙子，过了$t_i$秒后上方沙子的质量（期间沙漏有可能被翻来翻去）。

题解

离线后用线段树乱搞。
把要询问的$x_i$排序后依次扔进线段树中。先枚举$R_i$更新答案，然后把沙漏翻转。翻转操作其实相当于把每个节点加上$(R[i]-R[i-1])*f_i,(f_i=-f_{i-1})$，但为了把答案控制在$[1,X]$内，若线段树当前节点的子树最小值+$(R[i]-R[i-1])*f$大于$X$，就是把子树全标记为$X$，小于0同理。（详见代码）
好像还有一种不用线段树的方法，然而我菜，并不会。。

代码

#pragma GCC optimize(3)
#include<iostream>
#include<iomanip>
#include<stack>
#include<queue>
#include<list>
#include<vector>
#include<set> 
#include<map>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#define ll long long
#define db double
#define inf 200009
#define infm 800009
#define INF (int)1e9
#define mod 1000000007LL
#define pi acos(-1)
#define rd(n) {n=0;char ch;int f=0;do{ch=getchar();if(ch=='-'){f=1;}}while(ch<'0'||ch>'9');while('0'<=ch&&ch<='9'){n=(n<<1)+(n<<3)+ch-48;ch=getchar();}if(f)n=-n;}
using namespace std;
int n,maxX,K;

struct edge{
    int tset,tadd;
    int minv,maxv;
}t[inf*4];
int val[inf];

void ST_set(int u,int x){
    t[u].tset=x;
    t[u].maxv=t[u].minv=x;
    t[u].tadd=0;
    return;
}

void ST_add(int u,int x){
    t[u].tadd+=x;
    t[u].maxv+=x;
    t[u].minv+=x;
    return;
}

void ST_pushdown(int u){
    int ls=u*2,rs=u*2+1;
    if (t[u].tset!=-1){
        ST_set(ls,t[u].tset);
        ST_set(rs,t[u].tset);
        t[u].tset=-1;
    }
    if (t[u].tadd){
        ST_add(ls,t[u].tadd);
        ST_add(rs,t[u].tadd);
        t[u].tadd=0;
    }
    return;
}

void ST_pushup(int u){
    t[u].minv=t[u*2].minv;
    t[u].maxv=t[u*2+1].maxv;
    return;
}

void ST_build(int u,int l,int r){
    t[u].tadd=0;
    t[u].tset=-1;
    if (l==r){
        t[u].maxv=t[u].minv=val[l];
        return;
    }
    int mid=(l+r)/2;
    ST_build(u*2,l,mid);
    ST_build(u*2+1,mid+1,r);
    ST_pushup(u);
    return;
}

void ST_modify(int u,int l,int r,int x){
    if (x==0){
        return;
    }
    if (t[u].minv+x>=maxX){
        ST_set(u,maxX);
        return;
    }
    if (t[u].maxv+x<=0){
        ST_set(u,0);
        return;
    }
    if (t[u].minv+x>=0 && t[u].maxv+x<=maxX){
        ST_add(u,x);
        return;
    }
    ST_pushdown(u);
    int mid=(l+r)/2;
    ST_modify(u*2,l,mid,x);
    ST_modify(u*2+1,mid+1,r,x);
    ST_pushup(u);
    return;
}

int ST_query(int u,int l,int r,int k){
    if (l==r){
        return t[u].maxv;
    }
    ST_pushdown(u);
    int mid=(l+r)/2;
    if (k<=mid){
        return ST_query(u*2,l,mid,k);
    }
    else{
        return ST_query(u*2+1,mid+1,r,k);
    }
    ST_pushup(u);
    return 0;
}


struct data{
    int t,a,id;
}a[inf];
int R[inf],rnk[inf];

bool cmp1(data _1,data _2){
    return _1.a<_2.a;
}

bool cmp2(data _1,data _2){
    return _1.t<_2.t;
}

int main(){
    rd(maxX) rd(K)
    for (int i=1;i<=K;i++){
        rd(R[i])
    }
    rd(n)
    for (int i=1;i<=n;i++){
        rd(a[i].t) rd(a[i].a)
        a[i].id=i;
    }
    R[++K]=a[n].t+1;
    sort(a+1,a+n+1,cmp1);
    for (int i=1;i<=n;i++){
        val[i]=a[i].a;
        rnk[a[i].id]=i;
    }
    sort(a+1,a+n+1,cmp2);
    ST_build(1,1,n);
    int f=-1,pos=0;
    for (int i=1;i<=K;i++){
        while (pos<n && a[pos+1].t<R[i]){
            pos++;
            int ans=ST_query(1,1,n,rnk[pos]);
            ans+=f*(a[pos].t-R[i-1]);
            ans=max(0,ans);
            ans=min(maxX,ans);
            printf("%d\n",ans);
        }
        ST_modify(1,1,n,(R[i]-R[i-1])*f);
        f=-f;
    }
    return 0;
}