luogu3960【noip2017】列队(动态开点splay)

许久的坑,在wc崩坏之后填一填。因为空间不够,所以要动态开点。每行前m-1个位置用一棵splay维护编号,最后一列用一棵splay维护。蒟蒻至今不会树状数组做法,还希望有dalao指教。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 300010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,q,sz[N*5],c[N*5][2],fa[N*5],rt[N],num=0,sz2[N*5];
ll v[N*5],ans;
inline void update(int p){
    int l=c[p][0],r=c[p][1];
    sz[p]=sz[l]+sz[r]+sz2[p];
}
inline void build(int &p,int l,int r){
    int mid=l+r>>1;p=mid;v[p]=(ll)(mid-1)*m;sz2[p]=1;
    if(l<mid) build(c[p][0],l,mid-1),fa[c[p][0]]=p;
    if(r>mid) build(c[p][1],mid+1,r),fa[c[p][1]]=p;
    update(p);
}
inline void rotate(int x,int &k){
    int y=fa[x],z=fa[y],l=x==c[y][1],r=l^1;
    if(y==k) k=x;
    else c[z][y==c[z][1]]=x;
    fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
    c[y][l]=c[x][r];c[x][r]=y;update(y);update(x); 
}
inline void splay(int x,int &k){
    while(x!=k){
        int y=fa[x],z=fa[y];
        if(y!=k){
            if(x==c[y][1]^y==c[z][1]) rotate(x,k);
            else rotate(y,k);
        }rotate(x,k);
    }
}
inline int find(int p,int x){
    if(x<=sz[c[p][0]]) return find(c[p][0],x);
    x-=sz[c[p][0]];
    if(x>sz2[p]) return find(c[p][1],x-sz2[p]);
    ans=x;return p;
}
inline void del(int x){
    splay(x,rt[0]);
    int pre=c[x][0],succ=c[x][1];
    while(c[pre][1]) pre=c[pre][1];
    while(c[succ][0]) succ=c[succ][0];
    splay(pre,rt[0]);splay(succ,c[pre][1]);
    c[succ][0]=0;fa[x]=0;update(succ);update(pre);
}
inline void ins(int &p,ll val,int Fa){
    if(p==n+2){ins(c[p][0],val,p);return;}
    if(!p){p=++num;v[num]=val;sz2[p]=1;fa[p]=Fa;update(p);return;}
    ins(c[p][1],val,p);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();q=read();build(rt[0],1,n+2);num=n+2;
    for(int i=1;i<=n;++i){
        rt[i]=++num;v[num]=(ll)(i-1)*m+1;sz2[num]=m-1;sz[num]=m-1;
    }while(q--){
        int x=read(),y=read();
        if(y==m){
            int xx=find(rt[0],x+1);ll res=v[xx];
            printf("%lld\n",res);
            del(xx);ins(rt[0],res,0);splay(num,rt[0]);continue;
        }int xx=find(rt[x],y);ll res=v[xx]+ans-1;
        printf("%lld\n",res);splay(xx,rt[x]);
        if(ans==sz2[xx]) sz2[xx]--,update(xx);
        else if(ans==1) sz2[xx]--,v[xx]++,update(xx);
        else{
            v[++num]=v[xx]+ans;sz2[num]=sz2[xx]-ans;
            sz2[xx]=ans-1;fa[c[xx][1]]=num;c[num][1]=c[xx][1];
            c[xx][1]=num;fa[num]=xx;update(num);update(xx);
        }xx=find(rt[0],x+1);ins(rt[x],v[xx],0);splay(num,rt[x]);
        del(xx);ins(rt[0],res,0);splay(num,rt[0]);
    }return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值