LOJ6283 数列分块入门7(分块)

pushdown的addtag[x]打成addtag[i],结果WA了一次

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
const int MOD = 10007;
int belong[100100],tagadd[100100],tagmul[100100],a[100100],sz,blocknum,n;
void calbe(int n){
    for(int i=1;i<=n;i++)
        belong[i]=(i-1)/sz+1;
}
void pushdown(int x){
    for(int i=sz*(x-1)+1;i<=min(sz*x,n);i++)
        a[i]=((long long)a[i]%MOD*tagmul[x]%MOD+tagadd[x]%MOD)%MOD;
    tagmul[x]=1;
    tagadd[x]=0;
}
void add(int l,int r,int c){
    int lsx=belong[l];
    int rex=belong[r];
    pushdown(lsx);
    for(int i=l;i<=min(lsx*sz,r);i++){
        a[i]=(a[i]%MOD+c%MOD)%MOD;
    }
    if(lsx!=rex){
        pushdown(rex);
        for(int i=(rex-1)*sz+1;i<=r;i++)
            a[i]=(a[i]%MOD+c%MOD)%MOD;
        for(int i=lsx+1;i<=rex-1;i++)
            tagadd[i]=(tagadd[i]%MOD+c%MOD)%MOD;
    }
}
void mul(int l,int r,int c){
    int lsx=belong[l];
    int rex=belong[r];
    pushdown(lsx);
    for(int i=l;i<=min(lsx*sz,r);i++)
        a[i]=((long long)a[i]%MOD*c%MOD)%MOD;
    if(lsx!=rex){
        pushdown(rex);
        for(int i=(rex-1)*sz+1;i<=r;i++)
            a[i]=((long long)a[i]%MOD*c%MOD)%MOD;
        for(int i=lsx+1;i<=rex-1;i++){
            tagadd[i]=((long long)tagadd[i]%MOD*c%MOD)%MOD;
            tagmul[i]=((long long)tagmul[i]%MOD*c%MOD)%MOD;
        }
    }
}
int query(int r){
    return ((long long)a[r]%MOD*tagmul[belong[r]]%MOD+tagadd[belong[r]]%MOD)%MOD;
}
int main(){
    // freopen("a5.in","r",stdin);
    // freopen("test.out","w",stdout);
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    sz=sqrt(n);
    blocknum=n/sz;
    if(n%sz)
        blocknum++;
    calbe(n);
    for(int i=1;i<=blocknum;i++)    
        tagmul[i]=1,tagadd[i]=0;
    for(int i=1;i<=n;i++){
        int opt,l,r,c;
        scanf("%d %d %d %d",&opt,&l,&r,&c);
        if(opt==0)
            add(l,r,c);
        else if(opt==1)
            mul(l,r,c);
        else
            printf("%d\n",query(r));
    } 
    return 0;
}
posted @ 2018-11-30 19:17  dreagonm  阅读(139)  评论(0编辑  收藏  举报