bzoj4942: [Noi2017]整数【线段树+压位】

本文链接：https://blog.csdn.net/cdsszjj/article/details/79857959

解题思路：

比较容易想到线段树维护二进制数的进退位，然后把加的操作也拆成二进制分log次加减，但这样是 $O(n*log_230n*log_21e9)$ ，显然过不了。
考虑压位，用一个int表示连续30位的01状态，加减也最多会涉及两个int，而且进退位除了这两个，其余只会加减1，也只用记录区间内是否全为 $2^{30}-1$ 或0即可。

#include<bits/stdc++.h>
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=4000005,s=(1<<30)-1;
int n;
int v0[N],v1[N],tag[N];
void pushdown(int k)
{
    if(tag[k]==-1)return;
    v0[k<<1]=v1[k<<1]=v0[k<<1|1]=v1[k<<1|1]=tag[k<<1]=tag[k<<1|1]=tag[k];
    tag[k]=-1;
}
void update(int k)
{
    v0[k]=v0[k<<1]|v0[k<<1|1];
    v1[k]=v1[k<<1]&v1[k<<1|1];
}
void add(int k,int l,int r,int p,int v)
{
    if(l==r){v0[k]+=v,v1[k]+=v;return;}
    pushdown(k);
    int mid=l+r>>1;
    if(p<=mid)add(k<<1,l,mid,p,v);
    else add(k<<1|1,mid+1,r,p,v);
    update(k);
}
void cover(int k,int l,int r,int x,int y,int v)
{
    if(x<=l&&r<=y){v0[k]=v1[k]=tag[k]=v;return;}
    pushdown(k);
    int mid=l+r>>1;
    if(x<=mid)cover(k<<1,l,mid,x,y,v);
    if(y>mid)cover(k<<1|1,mid+1,r,x,y,v);
    update(k);
}
int query(int k,int l,int r,int p)
{
    if(l==r)return v0[k];
    pushdown(k);
    int mid=l+r>>1;
    if(p<=mid)return query(k<<1,l,mid,p);
    else return query(k<<1|1,mid+1,r,p);
}
int find_s(int k,int l,int r,int p)
{
    if(v1[k]==s)return -1;
    if(l==r)return l;
    pushdown(k);
    int mid=l+r>>1;
    if(p<=mid)
    {
        int res=find_s(k<<1,l,mid,p);
        return ~res?res:find_s(k<<1|1,mid+1,r,p);
    }
    else return find_s(k<<1|1,mid+1,r,p);   
}
int find_0(int k,int l,int r,int p)
{
    if(!v0[k])return -1;
    if(l==r)return l;
    int mid=l+r>>1;
    pushdown(k);
    if(p<=mid)
    {
        int res=find_0(k<<1,l,mid,p);
        return ~res?res:find_0(k<<1|1,mid+1,r,p);
    }
    else return find_0(k<<1|1,mid+1,r,p);
}
void Add(int p,int v)
{
    if(!v)return;
    int t=query(1,0,n,p);
    if(t+v<=s)add(1,0,n,p,v);
    else 
    {
        add(1,0,n,p,v-s-1);
        int q=find_s(1,0,n,p+1);
        if(q!=p+1)cover(1,0,n,p+1,q-1,0);
        add(1,0,n,q,1);
    }
}
void Del(int p,int v)
{
    if(!v)return;
    int t=query(1,0,n,p);
    if(t>=v)add(1,0,n,p,-v);
    else
    {
        add(1,0,n,p,s+1-v);
        int q=find_0(1,0,n,p+1);
        if(q!=p+1)cover(1,0,n,p+1,q-1,s);
        add(1,0,n,q,-1);
    }
}
int main()
{
    //freopen("lx.in","r",stdin);
    n=getint();getint(),getint(),getint();
    int op,x,y,p;
    for(int i=1;i<=n;i++)
    {
        op=getint(),x=getint();
        if(op==1)
        {
            y=getint(),p=y/30;
            if(x>0)
            {
                Add(p,(x<<(y-p*30))&s);
                Add(p+1,x>>(30-(y-p*30)));
            }
            else if(x<0)
            {
                x=-x;
                Del(p,(x<<(y-p*30))&s);
                Del(p+1,x>>(30-(y-p*30)));
            }
        }
        else 
        {
            p=x/30;
            (query(1,0,n,p)&(1<<(x-p*30)))?puts("1"):puts("0");
        }
    }
    return 0;
}