[机房测试]WC

Description

西比拉系统正在观察一个人的心理。
这个人的心理可以抽象为一条数轴,上面存在一些非负整点,被称为「意识」。
观察的方式是选取一个非负整点作为采样点,尝试捕捉这个点附近的意识。
对于一次采样,西比拉系统会吸引在区间 \([l,r]\) 内的意识,这时意识会向采样点靠近,此处的 \([l,r]\) 称为这次采样的吸引区间。
每次采样又会有一个参数,捕捉半径 \(t\),表示如果有意识可以被吸引而移动到 \([x1-t,x1+t]\) 范围内,其数据就会被捕捉。其中 \(x1\) 表示采样点的位置。
注意,当一次采样结束,所有意识都会回到原来的位置。
每个意识会有一个参数,思维强度 \(s\),表示这个意识被吸引时的可移动范围为 \([x2-s,x2+s]\),其中 \(x2\) 表示这个意识出现的位置。
初始时数轴上没有意识,此后每个时刻仅会发生一个事件。
事件有两种:
·意识事件,形如 \(T \ x2 \ s\),表示在 \(x2\) 处出现了一个思维强度为 \(s\) 的意识。
·采样事件,形如 \(G \ x1 \ l \ r \ t\),表示在 \(x1\) 处进行一次吸引区间为 \([l,r]\),捕捉半径为 \(t\),请求出会被捕捉到的意识的个数。
对于每次采样,请求出会被捕捉到的意识的个数。

Solution

容易发现,就是需要统计满足条件

\[\begin{cases} |x_1-x_2|\leq t+s \\ l\leq x_2 \leq r \end{cases} \]

的点的个数。按照套路,把绝对值拆开,可以得到两种限制条件

\[\begin{cases} x_1<x_2\leq r \\ x_2-s\leq t+x_1 \end{cases} \text{or}\ \begin{cases} l\leq x_2 \leq x_1\\ x_2+s\geq x_1-t \end{cases} \]

加上时间限制,跑两遍CDQ分治即可,内层用平衡树查区间数的个数。

考场代码写得有点不一样,是把询问差分了,但是是一个道理。

#include<stdio.h>
#include<algorithm>
#include<stdlib.h>
using namespace std;

inline int read(){
    int x=0,flag=1; char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')flag=0;c=getchar();}
    while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+c-48;c=getchar();}
    return flag? x:-x;
}

const int N=2e5+7;

struct Que{
    int pos,x,s; bool op; int P;
    Que(int pos_=-1,int x_=0,int s_=0,bool op_=0,int P_=0):
        pos(pos_),x(x_),s(s_),op(op_),P(P_){}
}a[N];

struct Node{
    int sz,key,ls,rs,val;
}t[N];

int top=0,rev[N],tp=0,ans[N];

inline bool Cmp1(const Que &X,const Que &Y){
    int x=X.x,y=Y.x;
    if(~X.pos) x=min(x,X.pos);
    if(~Y.pos) y=min(y,Y.pos);
    return x<y;
}

inline bool Cmp2(const Que &X,const Que &Y){
    int x=(~X.pos)? X.x+X.s:X.x-X.s;
    int y=(~Y.pos)? Y.x+Y.s:Y.x-Y.s;
    return x<y;
}

void Del(int id){
    if(t[id].ls) Del(t[id].ls);
    if(t[id].rs) Del(t[id].rs);
    rev[++tp]=id;
}

inline int New(int x){
    static int cnt=0; int p;
    if(tp) p=rev[tp--];
    else p=++cnt;
    t[p]=(Node){1,rand(),0,0,x};
    return p;
}

inline void update(int id){
    t[id].sz=t[t[id].ls].sz+t[t[id].rs].sz+1;
}

int merge(int x,int y){
    if(!x||!y) return x+y;
    if(t[x].key<t[y].key){
        t[x].rs=merge(t[x].rs,y);
        return update(x),x;
    }else{
        t[y].ls=merge(x,t[y].ls);
        return update(y),y;
    }
}

void split(int id,int k,int &x,int &y){
    if(!id){x=y=0;return;}
    if(t[id].val<=k)
        x=id,split(t[id].rs,k,t[id].rs,y);
    else y=id,split(t[id].ls,k,x,t[id].ls);
    update(id);
}

void CDQ(int lf,int rf){
    if(lf==rf) return ;
    int mid=(lf+rf)>>1;
    CDQ(lf,mid),CDQ(mid+1,rf);
    sort(a+lf,a+mid+1,Cmp1),sort(a+mid+1,a+rf+1,Cmp1);
    int rt=0,x,y,z;
    for(int i=mid+1,j=lf;i<=rf;i++){
        if(a[i].pos==-1) continue;
        const int rg=min(a[i].pos,a[i].x);
        while(j<=mid&&((~a[j].pos)||(a[j].x<=rg))){
            if(~a[j].pos){j++;continue;}
            split(rt,a[j].x+a[j].s,x,y);
            rt=merge(merge(x,New(a[j].x+a[j].s)),y);
            j++;
        }
        split(rt,a[i].x-a[i].s-1,x,y);
        ans[a[i].P]+=(a[i].op? 1:-1)*t[y].sz;
        rt=merge(x,y);
    }
    if(rt) Del(rt); rt=0;
    sort(a+lf,a+mid+1,Cmp2),sort(a+mid+1,a+rf+1,Cmp2);
    for(int i=mid+1,j=lf;i<=rf;i++){
        if(a[i].pos==-1) continue;
        if(a[i].x>=a[i].pos) continue;
        const int rg=a[i].x+a[i].s;
        while(j<=mid&&((~a[j].pos)||(a[j].x-a[j].s<=rg))){
            if(~a[j].pos){j++;continue;}
            split(rt,a[j].x,x,y);
            rt=merge(merge(x,New(a[j].x)),y);
            j++;
        }
        split(rt,a[i].pos,y,z);
        split(y,a[i].x,x,y);
        ans[a[i].P]+=(a[i].op? 1:-1)*t[y].sz;
        rt=merge(merge(x,y),z);
    }
    if(rt) Del(rt);
}

int main(){
    freopen("mind.in","r",stdin);
    freopen("mind.out","w",stdout);
    srand(114514);
    int n=read(),q=0; char op[5];
    for(int i=1;i<=n;i++){
        scanf("%s",op);
        if(op[0]=='T'){
            int x=read()+1,s=read();
            a[++top]=Que(-1,x,s);
        }else{
            q++;
            int x=read()+1,l=read()+1,r=read()+1,s=read();
            a[++top]=Que(l-1,x,s,0,q),a[++top]=Que(r,x,s,1,q);
        }
    }
    CDQ(1,top);
    for(int i=1;i<=q;i++) printf("%d\n",ans[i]);
}
上一篇:二维差分(差分矩阵)


下一篇:Docker启动报错