【FOJ】1962 新击鼓传花游戏

#include<cstdio>
#define MAXN 500010
int tree[MAXN<<2];
inline void PushUp(int rt)
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void Build(int L,int R,int rt)
{
    if(L==R)
        tree[rt]=1;
    else
    {
        int mid=(L+R)>>1;
        Build(L,mid,rt<<1);
        Build(mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
void Leave(int k,int L,int R,int rt)
{
    if(L==R)
        tree[rt]=0;
    else
    {
        int mid=(L+R)>>1;
        if(tree[rt<<1]>=k)
            Leave(k,L,mid,rt<<1);
        else
            Leave(k-tree[rt<<1],mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
void Back(int x,int L,int R,int rt)
{
    if(L==R)
        tree[rt]=1;
    else
    {
        int mid=(L+R)>>1;
        if(mid>=x)
            Back(x,L,mid,rt<<1);
        else
            Back(x,mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
int Query(int k,int L,int R,int rt)
{
    if(L==R)
        return L;
    int mid=(L+R)>>1;
    if(tree[rt<<1]>=k)
        return Query(k,L,mid,rt<<1);
    else
        return Query(k-tree[rt<<1],mid+1,R,rt<<1|1);
}
int main()
{
    int n,m,k;
    char ch;
    while(~scanf("%d%d",&n,&m))
    {
        Build(1,n,1);
        while(m--)
        {
            scanf(" %c%d",&ch,&k);
            if(ch=='L')
                Leave(k,1,n,1);
            else if(ch=='R')
                Back(k,1,n,1);
            else
                printf("%d\n",Query(k,1,n,1));
        }
    }
    return 0;
}