bzoj4942 [Noi2017]整数(线段树)

每2^30压成一位。正好O(n)位,每一位还都是int以内。

模拟加减法。每次定位到相应位置上,我是取出了两位来做,正好不超ll很妙。

然后对于进位退位问题不能暴力搞,我们可以维护一个区间左起第一个非0/非1数。然后区间打标记修改。
细节较多。

#include 
#include 
#include 
#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 1000010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return  EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,bin[31],id;
struct node{
    int f[2],cov,x;//f0--最左非0,f1--最左非1,cov==0 cov==1 cov==-1
}tr[N<<2];
inline void docov(int p,int l,int val){
    tr[p].cov=val;if(val) tr[p].x=bin[30]-1;
    else tr[p].x=0;
    tr[p].f[val]=n+1;tr[p].f[val^1]=l;
}
inline void pushdown(int p,int l,int r,int mid){
    if(tr[p].cov==-1) return;
    docov(p<<1,l,tr[p].cov);docov(p<<1|1,mid+1,tr[p].cov);
    tr[p].cov=-1;
}
inline void pushup(int p){
    tr[p].f[0]=min(tr[p<<1].f[0],tr[p<<1|1].f[0]);
    tr[p].f[1]=min(tr[p<<1].f[1],tr[p<<1|1].f[1]);
}
inline void update(int p,int l){
    tr[p].f[0]=tr[p].f[1]=l;
    if(tr[p].x==0) tr[p].f[0]=n+1;
    if(tr[p].x==bin[30]-1) tr[p].f[1]=n+1;
}
inline ll ask(int p,int l,int r,int x){
    if(l==r) return tr[p].x;
    int mid=l+r>>1;pushdown(p,l,r,mid);
    if(x<=mid) return ask(p<<1,l,mid,x);
    return ask(p<<1|1,mid+1,r,x);
}
inline void change(int p,int l,int r,int x,int val){
    if(l==r){tr[p].x=val;update(p,l);return;}
    int mid=l+r>>1;pushdown(p,l,r,mid);
    if(x<=mid) change(p<<1,l,mid,x,val);
    else change(p<<1|1,mid+1,r,x,val);pushup(p);
}
inline void build(int p,int l,int r){
    tr[p].cov=-1;
    if(l==r){tr[p].f[0]=n+1;tr[p].f[1]=l;return;}
    int mid=l+r>>1;
    build(p<<1,l,mid);build(p<<1|1,mid+1,r);pushup(p);
}
inline void add(int p,int l,int r,int x,int val){
    if(l==r){tr[p].x+=val;update(p,l);return;}
    int mid=l+r>>1;pushdown(p,l,r,mid);
    if(x<=mid) add(p<<1,l,mid,x,val);
    else add(p<<1|1,mid+1,r,x,val);pushup(p);
}
inline void ask(int p,int l,int r,int x,int y,int val){//求最左非val
    if(x==l&&r==y){id=min(id,tr[p].f[val]);return;}
    int mid=l+r>>1;pushdown(p,l,r,mid);
    if(y<=mid) ask(p<<1,l,mid,x,y,val);
    else if(x>mid) ask(p<<1|1,mid+1,r,x,y,val);
    else ask(p<<1,l,mid,x,mid,val),ask(p<<1|1,mid+1,r,mid+1,y,val);
}
inline void cover(int p,int l,int r,int x,int y,int val){
    if(x>y) return;if(x<=l&&r<=y){docov(p,l,val);return;}
    int mid=l+r>>1;pushdown(p,l,r,mid);
    if(x<=mid) cover(p<<1,l,mid,x,y,val);
    if(y>mid) cover(p<<1|1,mid+1,r,x,y,val);pushup(p);
}
int main(){
//  freopen("a2.in","r",stdin);
//  freopen("a.out","w",stdout);
    n=m=read();int t1=read(),t2=read(),t3=read();n+=2;bin[0]=1;
    for(int i=1;i<=30;++i) bin[i]=bin[i-1]<<1;build(1,1,n);
    while(m--){
        int op=read();
        if(op==1){
            ll a=read();int b=read();
            int x=b/30+1,y=b%30;a*=bin[y];
            ll res=ask(1,1,n,x+1)*bin[30]+ask(1,1,n,x);
            if(a>=0){
                res+=a;change(1,1,n,x,res%bin[30]);res/=bin[30];
                change(1,1,n,x+1,res%bin[30]);res/=bin[30];
                if(!res) continue;
                id=n+1;ask(1,1,n,x+2,n,1);
                add(1,1,n,id,1);cover(1,1,n,x+2,id-1,0);
            }else{
                bool flag=0;res+=a;
                if(res<0) res+=(ll)bin[30]*bin[30],flag=1;
                change(1,1,n,x,res%bin[30]);res/=bin[30];
                change(1,1,n,x+1,res%bin[30]);if(!flag) continue;
                id=n+1;ask(1,1,n,x+2,n,0);
                add(1,1,n,id,-1);cover(1,1,n,x+2,id-1,1);
            }
        }else{
            int b=read();
            int x=b/30+1,y=b%30;
            puts(ask(1,1,n,x)&bin[y]?"1":"0");
        }
    }return 0;
}

你可能感兴趣的:(bzoj,线段树)