数列分块入门 7（LibreOj-6283）

【题目描述】

给出一个长为 n 的数列，以及 n 个操作，操作涉及区间乘法，区间加法，单点询问。

【输入格式】

第一行输入一个数字 n。

第二行输入 n 个数字，第 i 个数字为 ai，以空格隔开。

接下来输入 n 行询问，每行输入四个数字 opt、l、r、c，以空格隔开。

若 opt=0，表示将位于 [l,r] 的之间的数字都加 c。

若 opt=1，表示将位于 [l,r] 的之间的数字都乘 c。

若 opt=2，表示询问 ar 的值 mod 10007（l 和 c 忽略）。

【输出格式】

对于每次询问，输出一行一个数字表示答案。

【样例】

样例输入
7
1 2 2 3 9 3 2
0 1 3 1
2 1 3 1
1 1 4 4
0 1 7 2
1 2 6 4
1 1 6 5
2 2 6 4

样例输出
3
100

【数据范围与提示】

对于 100% 的数据，1<=n<=100000,-2^31<=other,ans<=2^31-1。

【源代码】

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<bitset>
#define EPS 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
const int MOD = 10000+7;
const int N = 1000000+5;
const int dx[] = {-1,1,0,0,-1,-1,1,1};
const int dy[] = {0,0,-1,1,-1,1,-1,1};
using namespace std;

int n;
int block,sum;//block为块的长度,sum为块的个数
int a[N];//存放数列元素
int pos[N];//pos记录第i个元素在第几个块中
int mulTag[N];//乘法标记
int addTag[N];//加法标记
void init(int n){
    block=sqrt(n);//块的长度
    sum=n/block;//块个数
    if(n%block)
        sum++;
    for(int i=1;i<=n;i++)//第i个元素在第几块中
        pos[i]=(i-1)/block+1;
    for(int i=1;i<=sum;i++)//乘法标记初始化为1
        mulTag[i]=1;
}
void reset(int x){//更新不完整块的值
    for(int i=(pos[x]-1)*block+1;i<=min(pos[x]*block,n);i++)//更新元素
        a[i]=(mulTag[pos[x]]*a[i]+addTag[pos[i]])%MOD;
    //更新标记
    mulTag[pos[x]]=1;
    addTag[pos[x]]=0;
}
void update(int L,int R,int x,int op){
    reset(L);//更新左端不完整块
    for(int i=L;i<=min(pos[L]*block,R);i++){//左边的边角料
        if(op==0)
            a[i]=(a[i]+x)%MOD;
        else
            a[i]=(a[i]*x)%MOD;
    }
    if(pos[L]!=pos[R]){//存在右区间才遍历，防止重复计算
        reset(R);//更新右端不完整块
        for(int i=(pos[R]-1)*block+1;i<=R;i++){//右边的边角料
            if(op==0)
                a[i]=(a[i]+x)%MOD;
            else
                a[i]=(a[i]*x)%MOD;
        }
    }
    for(int i=pos[L]+1;i<=pos[R]-1;i++){//中间的整块
        if(op==0)
            addTag[i]=(addTag[i]+x)%MOD;
        else{
            addTag[i]=(addTag[i]*x)%MOD;
            mulTag[i]=(mulTag[i]*x)%MOD;
        }
    }
}
int query(int x){
    return (a[x]*mulTag[pos[x]]+addTag[pos[x]])%MOD;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        a[i]%=MOD;
    }

    init(n);

    for(int i=1;i<=n;i++){
        int op;
        int left,right,x;
        scanf("%d",&op);
        scanf("%d%d%d",&left,&right,&x);
        if(op==0||op==1)//加或乘
            update(left,right,x,op);
        else if(op==2)
            printf("%d\n",query(right));
    }
    return 0;
}