洛谷P3960 列队（Splay）

传送门

 

感觉自己好久不打数据结构已经完全不会了orz……

据说正解树状数组？然而并不会

首先考虑一下每一次操作，就是把一个人从这一行中取出并放到行的最后，再从最后一列取出放到列的最后

那么这两种操作其实可以看做同一个类型，都是把某一个数取出并放到最后

那么这个可以用splay来搞，用splay维护区间，然后每一次找第k个相当于找第k大，然后删除之后在末尾插入

然后如果直接开空间怕是要炸……那只能缩点，等做到这个点的时候再把它split出来……

然后……看代码好了……

//minamoto
#include<cstdio>
#include<iostream>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
inline int read(){
    #define num ch-'0'
    char ch;bool flag=0;int res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
inline void print(ll x){
    if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
const int N=1e7+5,maxn=3e5+5;
int n,m,q,x,y;ll t,u;
int fa[N],ch[N][2],tot;ll L[N],R[N],sz[N];
struct Splay{
    #define ls ch[x][0]
    #define rs ch[x][1]
    int rt;
    inline int newnode(ll l,ll r){
        L[++tot]=l,R[tot]=r,sz[tot]=r-l+1;return tot;
    }
    inline void upd(int x){
        sz[x]=sz[ls]+sz[rs]+R[x]-L[x]+1;
    }
    void rotate(int x){
        int y=fa[x],z=fa[y],d=ch[y][1]==x;
        ch[z][ch[z][1]==y]=x;
        fa[x]=z,fa[y]=x,fa[ch[x][d^1]]=y,ch[y][d]=ch[x][d^1],ch[x][d^1]=y,upd(y);
    }
    void splay(int x){
        for(int y=fa[x],z=fa[y];fa[x];y=fa[x],z=fa[y]){
            if(fa[y])
            ((ch[y][1]==x)^(ch[z][1]==y))?rotate(x):rotate(y);
            rotate(x);
        }
        upd(x),rt=x;
    }
    inline void ins(ll v){
        int x=rt;
        while(rs) x=rs;
        newnode(v,v),fa[tot]=x,rs=tot,splay(tot);
    }
    ll split(int x,ll k){
        ll s=L[x],t=R[x];
        if(k!=s&&k!=t){
            if(s+1<t){
                int a=newnode(k+1,t);R[x]=k-1;
                ch[a][1]=ch[x][1],ch[x][1]=a,fa[a]=x,upd(a),upd(x);
                splay(a);
            }else{++L[x],splay(x);}
        }else{
            k==t?--t:++s;
            L[x]=s,R[x]=t,sz[x]=t-s+1,splay(x); 
        }
        return k;
    }
    ll rk(int k){
        int x=rt;
        while(true){
            if(sz[ls]>=k) x=ls;
            else{
                k-=sz[ls];
                if(k<=R[x]-L[x]+1) return split(x,k+L[x]-1);
                k-=R[x]-L[x]+1,x=rs;
            }
        }
    }
    inline void init(ll l,ll r){rt=newnode(l,r);}
    int build(int l,int r,int f){
        if(l>r) return 0;
        int mid=(l+r)>>1,x=newnode(1ll*m*mid,1ll*m*mid);
        fa[x]=f;
        ls=build(l,mid-1,x);
        rs=build(mid+1,r,x);
        return upd(x),x;
    }
}T[maxn];
int main(){
//    freopen("testdata.in","r",stdin);
    n=read(),m=read(),q=read();
    for(int i=1;i<=n;++i) T[i].init(1ll*(i-1)*m+1,1ll*i*m-1);
    T[0].rt=T[0].build(1,n,0);
    while(q--){
        x=read(),y=read();
        if(y==m){
            print(u=T[0].rk(x)),T[0].ins(u);
        }else{
            print(u=T[x].rk(y));
            t=T[0].rk(x);
            T[x].ins(t),T[0].ins(u);
        }
    }
    Ot();
    return 0;
}