P13979 数列分块入门 4题解

P13979 数列分块入门 4

题目背景

洛谷的数列分块入门系列的测试数据范围和原题有不同。

题目描述

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

输入格式

第一行输入一个数字 $n$。

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

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

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

若 $\mathrm{opt}=1$，表示询问位于 $[l,r]$ 的所有数字的和 $\mathrm{mod}(c+1)$。你需要输出非负的余数值。

输出格式

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

输入输出样例 #1

输入 #1

4
1 2 2 3
0 1 3 1
1 1 4 4
0 1 2 2
1 1 2 4

输出 #1

1
4

说明/提示

子任务

子任务 1（40 分）：$1\le n\le 50000,-2^{31}\le a_i,c\le 2^{31}-1$。

子任务 2（60 分）：$1\le n\le 300000,-2^{31}\le a_i,c\le 2^{31}-1$。

对于所有测试数据，满足 $1\le n\le 300000,-2^{31}\le a_i,c\le 2^{31}-1$。$1\le l\le r\le n$。$\mathrm{opt}\in \{0,1\},1\le l\le r\le n$。每次操作后的 $a_i$ 满足 $-2^{31}\le a_i\le 2^{31}-1$。特别地，数据保证当 $\mathrm{opt}=1$ 时，$c\ge 0$。

思路

直接分块，对于每个维护。

代码见下

#include<bits/stdc++.h>
using namespace std;
long long n,a[300005],d[300005],b[300005],lz[300005],wd[300005],ss=0,zs,l,r,c,ll[300005],rr[300005],op=0,li,ri,md,mid,op2=0;
int main(){
	cin>>n;
    ss=sqrt(n);
    for(int i=1;i<=n;i++){
        cin>>a[i];
        b[i]=((i-1)/ss)+1;
        if(ll[b[i]]==0){
            ll[b[i]]=i;
        }
        rr[b[i]]=i;
        zs=max(zs,b[i]);
        wd[b[i]]+=a[i];
    }
    for(int i=1;i<=n;i++){
        cin>>op>>l>>r>>c;
        //cout<<op<<endl;
        if(op==0){
            if(b[l]==b[r]){
                for(int j=l;j<=r;j++){
                    a[j]+=c;
                    wd[b[j]]+=c;
                }
            }
            else{
                for(int j=l;j<=rr[b[l]];j++){
                    a[j]+=c;
                    wd[b[j]]+=c;
                }
                for(int j=ll[b[r]];j<=r;j++){
                    a[j]+=c;
                    wd[b[j]]+=c;
                }
                for(int j=b[l]+1;j<=b[r]-1;j++){
                    lz[j]+=c;
                    //wd[b[j]]+=c*(rr[j]-ll[j]+1);
                }
            }
        }
        else{
            op2=0;
            //c*=c;
            //c++;
            if(b[l]==b[r]){
                for(int j=l;j<=r;j++){
                    op2+=(a[j]+lz[b[j]]);
                    //op2%=(c+1);
                }
                //sort(a+ll[b[l]],a+rr[b[l]]+1);
            }
            else{
                for(int j=l;j<=rr[b[l]];j++){
                    op2+=(a[j]+lz[b[j]]);
                    //op2%=(c+1);
                    //cout<<"1f"<<(d[j]+lz[b[j]])<<endl;
                }
                for(int j=ll[b[r]];j<=r;j++){
                    op2+=(a[j]+lz[b[j]]);
                    //%=(c+1);
                    //cout<<"1f"<<(d[j]+lz[b[j]])<<endl;
                }
                for(int j=b[l]+1;j<=b[r]-1;j++){
                    op2+=(wd[j]+lz[j]*(rr[j]-ll[j]+1));
                    //%=(c+1);
                    //cout<<"1f"<<wd[j]<<endl;
                }
            }  
            op2%=(c+1);
            cout<<(op2+c+1)%(c+1)<<endl;
        }
    }
	return 0;
}