【权值分块】bzoj1503 [NOI2004]郁闷的出纳员

权值分块，离散化非常蛋疼，只能离散化搞……

需要支持操作：删除<=某个值得所有权值==打标记 O(sqrt(n))

码长和我的平衡树差不多……速度快3倍左右。

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
#define N 201001
struct Point{int v,p;}tmp[N];
bool operator < (const Point &a,const Point &b){return a.v<b.v;}
int n,m,Infu,a[N],c[N],leave,
en/*插入的权值数*/,en2/*离散化之后的权值种类数*/,ma[N];
char op[N];
int num[N],l[460],CH[12],r[460],Num,sumv[460],sum,sz,b[N],all;
bool delta[N];
inline void R(int &x){
    char c=0;int 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()
{
    sz=sqrt(en2); if(!sz) sz=1;
    for(sum=1;sum*sz<en2;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]=en2;
    for(int i=l[sum];i<=r[sum];i++) num[i]=sum;
}
void pushdown(const int &p)
{
    if(delta[p])
      {
          for(int i=l[p];i<=r[p];i++) b[i]=0;
          delta[p]=0;
      }
}
inline void Insert(const int &x){pushdown(num[x]); b[x]++; sumv[num[x]]++; all++;}
inline void Delete(const int &v)//删除小于等于v的所有权值 
{
    int used=all; pushdown(num[v]);
    for(int i=v;i>=l[num[v]];i--)
      {
          sumv[num[i]]-=b[i];
          all-=b[i];
          b[i]=0;
      }
    for(int i=num[v]-1;i>=1;i--) if(sumv[i])
      {
          delta[i]=1;
          all-=sumv[i];
          sumv[i]=0;
      } leave+=(used-all);
}
inline int Kth(const int &x)
{
    int cnt=0;
    for(int i=sum;;i--)
      {
        cnt+=sumv[i];
        if(cnt>=x)
          {
            cnt-=sumv[i];
            for(int j=r[i];;j--)
            {cnt+=b[j]; if(cnt>=x) return j;}
          }
      }
}
int main()
{
    R(n); R(m);
    for(int i=1;i<=n;i++)
      {
          op[i]=getchar(); R(a[i]);
          if(op[i]=='I')
            {
                tmp[++en].v=a[i]-Infu;//为新员工消除之前工资变化的影响 
                tmp[en].p=en;
            }
          else if(op[i]=='A') Infu+=a[i];
          else if(op[i]=='S')
          {
              Infu-=a[i];
              tmp[++en].v=m-Infu-1;//每次删除<=m-Infu-1的权值 
              tmp[en].p=en;
          }
      }
    sort(tmp+1,tmp+en+1);
    ma[c[tmp[1].p]=++en2]=tmp[1].v;
    for(int i=2;i<=en;i++)
      {
          if(tmp[i].v!=tmp[i-1].v) en2++;
          ma[c[tmp[i].p]=en2]=tmp[i].v;
      } Infu=en=0; makeblock();
    for(int i=1;i<=n;i++)
      {
          if(op[i]=='I') {en++; if(a[i]>=m) Insert(c[en]);}
          else if(op[i]=='A') Infu+=a[i];
          else if(op[i]=='S')
          {
            Infu-=a[i];
            Delete(c[++en]);
          }
          else
            {
                if(a[i]>all) puts("-1");
                else P(ma[Kth(a[i])]+Infu);
            }
      } P(leave);
    return 0;
}