loj #6280. 數列分塊入門 4

#6280. 數列分塊入門 4

題目描述

給出一個長爲 nnn 的數列，以及 nnn 個操做，操做涉及區間加法，區間求和。

輸入格式

第一行輸入一個數字 nnn。

第二行輸入 nnn 個數字，第 i 個數字爲 aia_ia​i​​，以空格隔開。

接下來輸入 nnn 行詢問，每行輸入四個數字 opt\mathrm{opt}opt、lll、rrr、ccc，以空格隔開。

若 opt=0\mathrm{opt} = 0opt=0，表示將位於 [l,r][l, r][l,r] 的之間的數字都加 ccc。

若 opt=1\mathrm{opt} = 1opt=1，表示詢問位於 [l,r][l, r][l,r] 的全部數字的和 mod(c+1)mod (c+1)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% 100\%100% 的數據，1≤n≤50000,−231≤others 1 \leq n \leq 50000, -2^{31} \leq \mathrm{others}1≤n≤50000,−2​31​​≤others、ans≤231−1 \mathrm{ans} \leq 2^{31}-1ans≤2​31​​−1。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn 50010
using namespace std;
int n,block,pos[maxn],c;
long long add[maxn],sum[maxn],a[maxn];
int findl(int i){return (i-1)*block+1;}
int findr(int i){return min(i*block,n);}
void set(int id){
    int l=findl(id),r=findr(id);
    sum[id]=0;
    for(int i=l;i<=r;i++)
        a[i]+=add[id],sum[id]+=a[i];
    add[id]=0;
}
void modify(int l,int r,int c){
    for(int i=l;i<=min(findr(pos[l]),r);i++)a[i]+=c;set(pos[l]);
    if(pos[l]==pos[r])return;
    for(int i=findl(pos[r]);i<=r;i++)a[i]+=c;set(pos[r]);
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
        add[i]+=c;
}
int query(int l,int r,int c){
    int res=0;
    for(int i=l;i<=min(findr(pos[l]),r);i++)
        res=(res+(a[i]+add[pos[i]])%(c+1))%(c+1);
    if(pos[l]==pos[r])return res;
    for(int i=findl(pos[r]);i<=r;i++)
        res=(res+(a[i]+add[pos[i]])%(c+1))%(c+1);
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
        res=(res+(1LL*(findr(i)-findl(i)+1)*add[i]%(c+1)+sum[i])%(c+1))%(c+1);
    return res;
}
int main(){
    scanf("%d",&n);
    block=sqrt(n);
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=n;i++)pos[i]=(i-1)/block+1;
    for(int i=1;i<=pos[n];i++)set(i);
    int op,l,r,c;
    for(int i=1;i<=n;i++){
        scanf("%d%d%d%d",&op,&l,&r,&c);
        if(op==0)modify(l,r,c);
        else printf("%d\n",query(l,r,c));
    }
    return 0;
}