数列分块入门 4（LibreOj-6280）

【题目描述】

给出一个长为 n 的数列，以及 n 个操作，操作涉及区间加法，区间求和。

【输入格式】

第一行输入一个数字 n。

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

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

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

若 opt=1，表示询问位于 [l,r] 的所有数字的和 mod(c+1)。

【输出格式】

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

【样例】

样例输入
4
1 2 2 3
0 1 3 1
1 1 4 4
0 1 2 2
1 1 2 4

样例输出
1
4

【数据范围与提示】

对于 100% 的数据，1<=n<=50000,-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 = 1E9+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;

LL block,sum;//block为块的长度,sum为块的个数
LL a[N];//存放数列元素
LL pos[N],tag[N];//pos记录第i个元素在第几个块中,tag为操作标记
LL ans[N];//维护整块和
void init(LL n){
    block=sqrt(n);//块的长度
    sum=n/block;//块个数
    if(n%block)
        sum++;
    for(LL i=1;i<=n;i++)//第i个元素在第几块中
        pos[i]=(i-1)/block+1;
}
void work(LL L,LL R,LL x){//维护整块和
    LL start=0;
    for(LL i=L;i<=R;i++)
        start+=a[i];
    ans[x]=start;
}
void update(LL L,LL R,LL x){
    for(LL i=L;i<=min(pos[L]*block,R);i++){//左边的边角料
        ans[pos[i]]+=x;
        a[i]+=x;
    }
    if(pos[L]!=pos[R]){//存在右区间才遍历，防止重复计算
        for(LL i=(pos[R]-1)*block+1;i<=R;i++){//右边的边角料
            ans[pos[i]]+=x;
            a[i]+=x;
        }
    }
    for(LL i=pos[L]+1;i<=pos[R]-1;i++){//中间的整块
        tag[i]+=x;
    }
}
LL query(LL L,LL R,LL mod){
    LL res=0;
    for(LL i=L;i<=min(pos[L]*block,R);i++){//左边的边角料
        res+=a[i]+tag[pos[i]];
    }

    if(pos[L]!=pos[R]){//存在右区间才遍历，防止重复计算
        for(LL i=(pos[R]-1)*block+1;i<=R;i++){//右边的边角料
            res+=a[i]+tag[pos[i]];
        }
    }

    for(LL i=pos[L]+1;i<=pos[R]-1;i++){//中间的整块进行二分查找
        res+=ans[i]+tag[i]*block;
    }
    return res%mod;
}
int main(){
    LL n;
    scanf("%lld",&n);
    for(LL i=1;i<=n;i++)
        scanf("%lld",&a[i]);

    init(n);
    for(LL i=1;i<=sum;i++)//传入块的左右边界与编号
        work((i-1)*block+1,i*block,i);

    for(LL i=1;i<=n;i++){
        LL op;
        LL left,right,x;
        scanf("%lld",&op);
        scanf("%lld%lld%lld",&left,&right,&x);
        if(op==0)
            update(left,right,x);
        else if(op==1)
            printf("%lld\n",query(left,right,x+1));
    }
    return 0;
}