【权值分块】bzoj1861 [Zjoi2006]Book 书架

权值分块……rank3……没什么好说的。

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int n,sz,sum,x,y,l[501],r[501],Min,Max,sumv[501],num[250001],m,v[250001],p[250001];
bool b[250001];
char op[6],c;
int Num,CH[12],f;
inline void R(int &x){
    c=0;f=1;
    for(;c<'0'||c>'9';c=getchar())if(c=='-')f=-1;
    for(x=0;c>='0'&&c<='9';c=getchar())(x*=10)+=(c-'0');
    x*=f;
}
inline void P(int x)
{
    if(!x){putchar('0');puts("");return;}
    if(x<0){putchar('-');x=-x;}Num=0;
    while(x>0)CH[++Num]=x%10,x/=10;
    while(Num)putchar(CH[Num--]+48);puts("");
}
void makeblock(const int &LIMIT)
{
    sz=sqrt((double)LIMIT*0.9); if(!sz) sz=1;
    for(sum=1;sum*sz<LIMIT;++sum)
      {
        l[sum]=r[sum-1]+1;
        r[sum]=sum*sz;
        for(int i=l[sum];i<=r[sum];++i) num[i]=sum;
      }
    l[sum]=r[sum-1]+1;
    r[sum]=LIMIT;
    for(int i=l[sum];i<=r[sum];++i) num[i]=sum;
}
inline void Insert(const int &x){b[x]=1; ++sumv[num[x]];}
inline void Delete(const int &x){b[x]=0; --sumv[num[x]];}
inline int Next(const int &x)
{
    for(int i=x+1;i<=r[num[x]];++i) if(b[i]) return i;
    for(int i=num[x]+1;;++i) if(sumv[i])
      for(int j=l[i];;++j)
        if(b[j]) return j;
}
inline int Pre(const int &x)
{
    for(int i=x-1;i>=l[num[x]];--i) if(b[i]) return i;
    for(int i=num[x]-1;;--i) if(sumv[i])
      for(int j=r[i];;--j)
        if(b[j]) return j;
}
inline int Rank(const int &x)
{
    int cnt=0;
    for(int i=num[Min];i<num[x];++i) cnt+=sumv[i];
    for(int i=l[num[x]];i<x;++i) cnt+=b[i];
    return cnt;
}
inline int Kth(const int &x)
{
    int cnt=0;
    for(int i=num[Min];;++i)
      {
        cnt+=sumv[i];
        if(cnt>=x)
          {
            cnt-=sumv[i];
            for(int j=l[i];;++j)
            {cnt+=b[j]; if(cnt==x) return j;}
          }
      }
}
int main()
{
    R(n); R(m); makeblock(n+(m<<1));
    for(int i=1;i<=n;++i)
      {
        R(v[i+m]);
        Insert(p[v[i+m]]=i+m);
      } Min=1+m; Max=n+m;
    for(int i=1;i<=m;++i)
      {
        scanf("%s",op); R(x);
        if(op[0]=='T')
          {
            Delete(p[x]);
            v[p[x]=--Min]=x;
            Insert(Min);
          }
        else if(op[0]=='B')
          {
            Delete(p[x]);
            v[p[x]=++Max]=x;
            Insert(Max);
          }
        else if(op[0]=='I')
          {
            R(y); if(y==-1)
              {
                swap(v[p[x]],v[Pre(p[x])]);
                swap(p[x],p[v[p[x]]]);
              }
            else if(y==1)
              {
                swap(v[p[x]],v[Next(p[x])]);
                swap(p[x],p[v[p[x]]]);
              }
          }
        else if(op[0]=='A') P(Rank(p[x]));
        else P(v[Kth(x)]);
      }
    return 0;
}