數列分塊入門 1 LOJ6277

 

題目描述

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

輸入格式

第一行輸入一個數字 n。c++

第二行輸入 n 個數字,第 iii 個數字爲 ai​​,以空格隔開。ui

接下來輸入 n 行詢問,每行輸入四個數字 opt、l、r、c,以空格隔開。spa

若 opt=0,表示將位於 [l,r] 的之間的數字都加 c。code

若 opt=1,表示詢問 ar​​ 的值(ll和 c忽略)。htm

輸出格式

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

 

樣例

樣例輸入

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

樣例輸出

2
5

題解:

分塊寫法ip

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int MAXN=50000+10;

ll a[MAXN];
int block,num;  //塊大小 塊數量
int belong[MAXN],l[MAXN],r[MAXN];//歸屬那一塊,塊左右端點
ll sum[MAXN]; // 塊的信息修改
int n;
void build()
{
    block=sqrt(n);
    num=n/block;
    if(n%block!=0) num++; // 不能被整分。
    for (int i = 1; i <= num ; ++i) {
        l[i]=(i-1)*block+1;r[i]=i*block;
    }
    r[num]=n;
    for (int i = 1; i <=n ; ++i) {
        belong[i]=(i-1)/block+1;
    }
}
void updata(int x,int y,int c)
{
    for (int i = x; i <=min(r[belong[x]],y) ; ++i) {// 注意此處取二者最小值
        a[i]+=c;
    }
    if(belong[x]!=belong[y])
        for (int i = l[belong[y]]; i <=y ; ++i) {
            a[i]+=c;
        }
    for (int i = belong[x]+1; i <=belong[y]-1 ; ++i) {
        sum[i]+=c;
    }

}
ll ask(int x)
{
    return a[x]+sum[belong[x]];
}
int main()
{
    scanf("%d",&n);
    memset(sum,0, sizeof(sum));
    for (int i = 1; i <=n ; ++i) {
        scanf("%lld",&a[i]);
    }
    int op,x,y,c;
    build();
    for (int i = 1; i <=n ; ++i) {
        scanf("%d%d%d%d",&op,&x,&y,&c);
        if(op==0)
        {
            updata(x,y,c);
        } else
        {
            printf("%lld\n",ask(y));
        }
    }
    return 0;
}
相關文章
相關標籤/搜索