【NOIP2017】列队 Splay

原题走这里

鉴于蒟蒻的脑细胞不足,看不懂大佬们的树状数组离线解法,于是就只能用Splay了。

暂时先不考虑最右一列,只考虑n行的左m-1个元素。
给每一行都建一个大小为m-1的Splay明显是不现实的。

这里存在一个小小的心理陷阱,心理素质差的人比如我一看到300000*300000的方阵就被吓到了,就会下意识地认为修改规模和元素的个数相等,可能会到达300000*300000级别,然后就会瞬间失去思考能力,结果就挂了。

然而并不是这样,如果所有的操作平均分配的话,有可能某行只有一两次修改。例如可能某行前m-1个元素全程都只有一次修改,则修改前是:
(1 2 3 4 … m-2 m-1)
修改后变成了:
(1 2 3 4 … x-2 x-1) (x+1 x+2 … m-2 m-1) (?)
可以看出,一次修改后,绝大多数的元素依然是连续的,
我们就可以把连续的元素合并成一个节点。
这样的话,一次修改就是把某一个连续区间劈开,然后在末尾再增加一个元素,至多增加两个节点。
从而,每行都建一个Splay就变得可行了,然后再给最后一列单独建一个正常的Splay即可。

虽然有点慢有点傻,但是能过呀!

代码如下:

// luogu-judger-enable-o2
#include 
using namespace std;
#define LL long long
#define f(x) t[x].f
#define s(x) t[x].s
#define l(x) t[x].l
#define d(x) t[x].d
#define c(x,y) t[x].c[y]
inline int read() {
    int x=0;
    char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return x;
}
struct data {
    int s,l,c[2],f;
    LL d;
    data(LL _d=0,int _l=0,int _f=0) {
        c[0]=c[1]=0;
        f=_f;
        d=_d;
        s=l=_l;
    }
} t[1800010];
int n,m,q,top,root[300010];
void update(int x) {
    s(x)=s(c(x,0))+s(c(x,1))+l(x);
}
inline void rotate(int x) {
    int y=f(x),z=f(f(x)),d=c(y,0)==x;
    f(x)=z;
    if(z)c(z,c(z,1)==y)=x;
    c(y,d^1)=c(x,d);
    f(c(x,d))=y;
    c(x,d)=y;
    f(y)=x;
    update(y);
    update(x);
}
void splay(int &w,int rt,int x) {
    while(f(x)!=rt) {
        int y=f(x),z=f(y);
        if(z!=rt)rotate((c(z,1)==y)^(c(y,1)==x)?x:y);
        rotate(x);
    }
    if(rt==0)w=x;
}
int search(int x,int y) {
    int i=root[x];
    while(1) {
        if(y<=s(c(i,0))) {
            i=c(i,0);
        } else if(y>s(c(i,0))+l(i)) {
            y-=s(c(i,0))+l(i);
            i=c(i,1);
        } else break;
    }
    return i;
}
LL query(int x,int y) {
    int temp=search(x,y);
    splay(root[x],0,temp);
    return d(temp)+y-s(c(temp,0))-1;
}
void insert(int &rt,int x,LL y,int z=1) {
    int i=rt;
    t[++top]=data(y,z);
    if(!rt) {
        rt=top;
        return;
    }
    while(i) {
        if(x<=s(c(i,0))) {
            if(c(i,0)) {
                i=c(i,0);
            } else {
                c(i,0)=top;
                break;
            }
        } else {
            x-=s(c(i,0))+l(i);
            if(c(i,1)) {
                i=c(i,1);
            } else {
                c(i,1)=top;
                break;
            }
        }
    }
    t[top].f=i;
    splay(rt,0,top);
}
void del(int x,int y) {
    int tp=search(x,y),t1,t2,i;
    splay(root[x],0,tp);
    i=c(root[x],1);
    t1=s(c(tp,0));
    t2=l(tp);
    if(i) {
        for(; c(i,0); i=c(i,0));
        splay(root[x],0,i);
        c(i,0)=c(c(i,0),0);
        f(c(i,0))=i;
        update(i);
    } else {
        root[x]=c(root[x],0);
        f(root[x])=0;
    }
    if(t1+11);
    if(y1,d(tp)+y-t1,t1+t2-y);
}
void init() {
    for(int i=1; i<=n; i++) {
        insert(root[i],s(root[i]),1LL*(i-1)*m+1,m-1);
        insert(root[0],s(root[0]),1LL*i*m);
    }
}
int main() {
    cin>>n>>m>>q;
    init();
    for(int i=1; i<=q; i++) {
        int x=read(),y=read();
        LL temp;
        if(y==m) {
            y=x;
            x=0;
        }
        temp=query(x,y);
        del(x,y);
        printf("%lld\n",temp);
        if(x!=0) {
            insert(root[x],s(root[x]),query(0,x));
            del(0,x);
        }
        insert(root[0],s(root[0]),temp);
    }
    return 0;
}

你可能感兴趣的:(【NOIP2017】列队 Splay)