loj #6277. 數列分塊入門 1

#6277. 數列分塊入門 1

題目描述

給出一個長爲 nnn 的數列，以及 nnn 個操做，操做涉及區間加法，單點查值。

輸入格式

第一行輸入一個數字 nnn。

第二行輸入 nnn 個數字，第 iii 個數字爲 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，表示詢問 ara_ra​r​​ 的值（lll 和 ccc 忽略）。

輸出格式

對於每次詢問，輸出一行一個數字表示答案。

樣例

樣例輸入

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

樣例輸出

2
5

數據範圍與提示

對於 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}-1 ans≤2​31​​−1。

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 50010
using namespace std;
int n,sum[maxn*4],lazy[maxn*4];
void build(int k,int l,int r){
    if(l==r){
        scanf("%d",&sum[k]);
        return;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);build(k<<1|1,mid+1,r);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}
void pushdown(int k,int l,int r){
    if(!lazy[k])return;
    int mid=(l+r)>>1;
    lazy[k<<1]+=lazy[k];sum[k<<1]+=lazy[k]*(mid-l+1);
    lazy[k<<1|1]+=lazy[k];sum[k<<1|1]+=lazy[k]*(r-mid);
    lazy[k]=0;
}
void modify(int k,int l,int r,int opl,int opr,int opv){
    if(l>=opl&&r<=opr){
        lazy[k]+=opv;
        sum[k]+=opv*(r-l+1);
        return;
    }
    pushdown(k,l,r);
    int mid=(l+r)>>1;
    if(opl<=mid)modify(k<<1,l,mid,opl,opr,opv);
    if(opr>mid)modify(k<<1|1,mid+1,r,opl,opr,opv);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}
int query(int k,int l,int r,int opx){
    if(l==r)return sum[k];
    pushdown(k,l,r);
    int mid=(l+r)>>1,res;
    if(opx<=mid)res=query(k<<1,l,mid,opx);
    else res=query(k<<1|1,mid+1,r,opx);
    sum[k]=sum[k<<1]+sum[k<<1|1];
    return res;
}
int main(){
    scanf("%d",&n);
    build(1,1,n);
    int op,l,r,c;
    for(int i=1;i<=n;i++){
        if(i==3){
            int x=1;
        }
        scanf("%d%d%d%d",&op,&l,&r,&c);
        if(op==0)modify(1,1,n,l,r,c);
        else printf("%d\n",query(1,1,n,r));
    }
    return 0;
}

線段樹區間修改+單點查詢

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn 50010
using namespace std;
int n,block,a[maxn],b[maxn],pos[maxn];
void update(int l,int r,int c){
    for(int i=l;i<=min(pos[l]*block,r);i++)a[i]+=c;
    if(pos[l]!=pos[r])
        for(int i=(pos[r]-1)*block+1;i<=r;i++)
        a[i]+=c;
    for(int i=pos[l]+1;i<=pos[r]-1;i++)b[i]+=c;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    block=sqrt(n);
    for(int i=1;i<=n;i++)pos[i]=(i-1)/block+1;
    int op,l,r,c;
    for(int i=1;i<=n;i++){
        scanf("%d%d%d%d",&op,&l,&r,&c);
        if(op==0)update(l,r,c);
        else printf("%d\n",a[r]+b[pos[r]]);
    }
    return 0;
}

分塊