LOJ 2302 「NOI2017」整数——压位线段树

题目：https://loj.ac/problem/2302

压30位，a最多落在两个位置上，拆成两次操作。

该位置加了 a 之后，如果要进位或者借位，查询一下连续一段 0 / 1 ，修改掉，再在含有 1 / 0 的那个位置上 -1 或者 +1 。

注意是在那个位置上 -1 或者 +1 而不是 -lowbit 或者 +lowbit 。

询问都是 <=30n ，所以只维护 30n 的范围即可。注意线段树压 30 位，开 n 个位置恰好是 0*n ~ 29*n，所以开 n+1 个位置。

线段树只需维护自己区间是全 0 / 1 还是都有。找连续一段 0/1 就在线段树上二分即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ls Ls[cr]
#define rs Rs[cr]
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
int Mx(int a,int b){return a>b?a:b;}
int Mn(int a,int b){return a<b?a:b;}
const int N=2e6+,bs=;
int n,nr,tot,Ls[N],Rs[N],lx[N],vl[N],tg[N];
int bin[bs+],U;  bool fx;
void build(int l,int r,int cr)
{
  tg[cr]=-; lx[cr]=vl[cr]=;
  if(l==r)return; int mid=l+r>>;
  ls=++tot; build(l,mid,ls);
  rs=++tot; build(mid+,r,rs);
}
void pshd(int cr)
{
  if(tg[cr]==-)return; int k=tg[cr]; tg[cr]=-;
  tg[ls]=tg[rs]=k;
  if(k==){ vl[ls]=vl[rs]=U; lx[ls]=lx[rs]=;}
  else { vl[ls]=vl[rs]=; lx[ls]=lx[rs]=;}
}
void pshp(int cr)
{
  if((!lx[ls])&&(!lx[rs]))lx[cr]=;
  else if(lx[ls]==&&lx[rs]==)lx[cr]=; else lx[cr]=;
}
void updt(int cr)
{ if(!vl[cr])lx[cr]=; else if(vl[cr]==U)lx[cr]=; else lx[cr]=;}
int fnd(int l,int r,int cr,int L,int R,int k)
{
  if(l>=L&&r<=R)
    {
      if(lx[cr]==(!k))return nr+; if(l==r)return l;
      int mid=l+r>>; pshd(cr);
      if(lx[ls]!=(!k))return fnd(l,mid,ls,L,R,k);
      return fnd(mid+,r,rs,L,R,k);
    }
  int mid=l+r>>,d=nr+; pshd(cr);
  if(L<=mid)d=fnd(l,mid,ls,L,R,k);
  if(d==nr+)d=fnd(mid+,r,rs,L,R,k);
  return d;
}
void mdfy(int l,int r,int cr,int L,int R,int k)
{
  if(l>=L&&r<=R)
    {
      tg[cr]=k; if(k==){vl[cr]=U;lx[cr]=;}
      else {vl[cr]=;lx[cr]=;} return;
    }
  int mid=l+r>>; pshd(cr);
  if(L<=mid)mdfy(l,mid,ls,L,R,k);
  if(mid<R)mdfy(mid+,r,rs,L,R,k);
  pshp(cr);
}
void mdfy2(int l,int r,int cr,int p,int k)
{
  if(l==r)
    {
      if(k==)vl[cr]++; else vl[cr]--;/////
      /*if(k==1){ int tmp=(vl[cr]^U); vl[cr]+=(tmp&(-tmp));}
    else vl[cr]-=(vl[cr]&(-vl[cr]));*/
      updt(cr); return;
    }
  int mid=l+r>>; pshd(cr);
  if(p<=mid)mdfy2(l,mid,ls,p,k);
  else mdfy2(mid+,r,rs,p,k);
  pshp(cr);
}
void add(int p)
{
  int d=fnd(,nr,,p,nr,);//fnd first 0
  if(p<d)mdfy(,nr,,p,d-,);
  if(d<=nr)mdfy2(,nr,,d,);//chg 0 to 1
}
void dec(int p)
{
  int d=fnd(,nr,,p,nr,);
  if(p<d)mdfy(,nr,,p,d-,);
  if(d<=nr)mdfy2(,nr,,d,);
}
void solve(int l,int r,int cr,int p,int k)
{
  if(l==r)
    {
      if(!fx)vl[cr]+=k; else vl[cr]-=k;
      if(vl[cr]>=bin[bs]){ vl[cr]-=bin[bs];add(l+);}
      if(vl[cr]<){ vl[cr]+=bin[bs];dec(l+);}
      updt(cr); return;
    }
  int mid=l+r>>; pshd(cr);
  if(p<=mid)solve(l,mid,ls,p,k); else solve(mid+,r,rs,p,k);
  pshp(cr);
}
bool qry(int l,int r,int cr,int p,int p2)
{
  if(l==r)return vl[cr]&bin[p2];
  int mid=l+r>>; pshd(cr);
  if(p<=mid)return qry(l,mid,ls,p,p2);
  else return qry(mid+,r,rs,p,p2);
}
int main()
{
  n=rdn(); nr=n+; int tp=rdn();tp=rdn();tp=rdn();
  bin[]=;for(int i=;i<=bs;i++)bin[i]=bin[i-]<<;
  U=bin[bs]-; tot=;build(,nr,);
  for(int i=,op,a,b,l,r;i<=n;i++)
    {
      op=rdn();
      if(op==)
    {
      a=rdn();b=rdn(); l=b; r=b+bs-;
      l=l/bs+; r=r/bs+;
      if(a<)a=-a,fx=; else fx=;
      if(l<r)
        {
          int k=a&(bin[l*bs-b]-);
          a>>=(l*bs-b); k<<=(b-(l-)*bs);
          solve(,nr,,l,k); solve(,nr,,r,a);
        }
      else solve(,nr,,l,a);
    }
      else
    {
      a=rdn(); l=a/bs+; a-=(l-)*bs;
      printf("%d\n",qry(,nr,,l,a));
    }
    }
  return ;
}