LOJ#2302. 「NOI2017」整数

$n \leq 1000000$个操作：一，给$x$加上$a*2^b$；二，问$x$的某个二进制位$k$。$b,k \leq 30n$，$|a| \leq 1e9$。

30暴露了一切。。可以把30个二进制位压一位，进位用线段树找到第一个0，而退位用类似的方法找到第一个1。

但其实第$k$位只由加的总量和减的总量的0到$k$这些数位上决定。因此可以把加减分成两个数组，不用再写一个线段树里的减法。回答时查一下$0$到$k-1$中加的和减的孰大孰小，以及第$k$位是否相同，分类可得答案。

这里写的直接模拟。

 //#include<iostream>
 #include<cstring>
 #include<cstdio>
 //#include<math.h>
 //#include<set>
 //#include<queue>
 //#include<bitset>
 //#include<vector>
 #include<algorithm>
 #include<stdlib.h>
 using namespace std;

 #define LL long long
 int qread()
 {
     char c; int s=,f=; while ((c=getchar())<'' || c>'') (c=='-') && (f=-);
     do s=s*+c-''; while ((c=getchar())>='' && c<=''); return s*f;
 }

 //Pay attention to '-' , LL and double of qread!!!!

 int m,n;
 #define maxn 1000011
 int pos[maxn];
 struct SMT
 {
     struct Node{int ls,rs,v; bool all0,all1,be0,be1;}a[maxn<<];
     int size,n;
     void up(int x)
     {a[x].all0=a[a[x].ls].all0&a[a[x].rs].all0; a[x].all1=a[a[x].ls].all1&a[a[x].rs].all1;}
     void build(int &x,int L,int R)
     {
         x=++size;
         a[x].all0=; a[x].all1=a[x].be0=a[x].be1=;
         if (L==R) {a[x].v=; pos[L]=x; return;}
         int mid=(L+R)>>;
         build(a[x].ls,L,mid); build(a[x].rs,mid+,R);
     }
     void clear(int N) {n=N; size=; int x; build(x,,n);}

     void besingle(int x,bool t)
     {
         a[x].all0=a[x].be0=!t;
         a[x].all1=a[x].be1=t;
         if (t) a[x].v=(<<)-; else a[x].v=;
     }
     void down(int x)
     {
         Node &b=a[x];
         if (b.be0) {besingle(b.ls,); besingle(b.rs,); b.be0=;}
         if (b.be1) {besingle(b.ls,); besingle(b.rs,); b.be1=;}
     }

     bool Add(int x,int L,int R,int p,int v)
     {
         if (L==R)
         {
             a[x].v+=v; if (a[x].v>=(<<))
             {
                 a[x].v^=(<<);
                 a[x].all0=(a[x].v==),a[x].all1=(a[x].v==(<<)-);
                 return ;
             }
             a[x].all0=(a[x].v==),a[x].all1=(a[x].v==(<<)-);
             return ;
         }
         down(x);
         int mid=(L+R)>>;
         if (p<=mid) {bool flag=Add(a[x].ls,L,mid,p,v); up(x); return flag;}
         bool flag=Add(a[x].rs,mid+,R,p,v); up(x); return flag;
     }
     bool f0(int x,int L,int R,int p)
     {
         if (L==R)
         {
             bool flag=;
             for (int i=;i<;i++) if (((a[x].v>>i)&)==) {a[x].v|=<<i; flag=; break;}
             else a[x].v^=(<<i);
             a[x].all0=(a[x].v==),a[x].all1=(a[x].v==(<<)-);
             return flag;
         }
         down(x);
         int mid=(L+R)>>;
         if (L>p)
         {
             if (a[x].all1) {besingle(x,); return ;}
             if (!f0(a[x].ls,L,mid,p)) f0(a[x].rs,mid+,R,p); up(x);
             return ;
         }
         if (mid>p)
         {
             if (f0(a[x].ls,L,mid,p)) {up(x); return ;}
             bool flag=f0(a[x].rs,mid+,R,p); up(x); return flag;
         }
         if (f0(a[x].rs,mid+,R,p)) {up(x); return ;}
         up(x); return ;
     }
     void add(int p,int v) {p++; if (Add(,,n,p,v) && p!=n) f0(,,n,p);}

     bool Minus(int x,int L,int R,int p,int v)
     {
         if (L==R)
         {
             a[x].v-=v; if (a[x].v<)
             {
                 a[x].v+=(<<);
                 a[x].all0=(a[x].v==),a[x].all1=(a[x].v==(<<)-);
                 return ;
             }
             a[x].all0=(a[x].v==),a[x].all1=(a[x].v==(<<)-);
             return ;
         }
         down(x);
         int mid=(L+R)>>;
         if (p<=mid) {bool flag=Minus(a[x].ls,L,mid,p,v); up(x); return flag;}
         bool flag=Minus(a[x].rs,mid+,R,p,v); up(x); return flag;
     }
     bool f1(int x,int L,int R,int p)
     {
         if (L==R)
         {
             bool flag=;
             for (int i=;i<;i++) if (((a[x].v>>i)&)==) {a[x].v^=<<i; flag=; break;}
             else a[x].v^=(<<i);
             a[x].all0=(a[x].v==),a[x].all1=(a[x].v==(<<)-);
             return flag;
         }
         down(x);
         int mid=(L+R)>>;
         if (L>p)
         {
             if (a[x].all0) {besingle(x,); return ;}
             if (!f1(a[x].ls,L,mid,p)) f1(a[x].rs,mid+,R,p);
             up(x); return ;
         }
         if (mid>p)
         {
             if (f1(a[x].ls,L,mid,p)) {up(x); return ;}
             bool flag=f1(a[x].rs,mid+,R,p); up(x); return flag;
         }
         if (f1(a[x].rs,mid+,R,p)) {up(x); return ;}
         up(x); return ;
     }
     void minus(int p,int v) {p++; if (Minus(,,n,p,v) && p!=n) f1(,,n,p);}

     int Query(int x,int L,int R,int p,int v)
     {
         if (L==R) return (a[x].v>>v)&;
         down(x);
         int mid=(L+R)>>;
         if (p<=mid) return Query(a[x].ls,L,mid,p,v);
         return Query(a[x].rs,mid+,R,p,v);
     }
     int query(int x) {return Query(,,n,x/+,x%);}
 }t;

 int main()
 {
     m=qread(); n=qread(); n=qread(); n=qread();
     n=;
     t.clear(n);

     int op,x,y;
     while (m--)
     {
         op=qread();
         if (op==)
         {
             x=qread(); y=qread();
             if (x>)
             {
                 int p=y/,q=(p+)*-y;
                 int a=x&((<<q)-),b=x>>q; a<<=y-p*;
                 t.add(p,a); if (b) t.add(p+,b);
             }
             else
             {
                 x=-x;
                 int p=y/,q=(p+)*-y;
                 int a=x&((<<q)-),b=x>>q; a<<=y-p*;
                 t.minus(p,a); if (b) t.minus(p+,b);
             }
         }
         else
         {
             x=qread();
             printf("%d\n",t.query(x));
         }
     }
     return ;
 }