1. 程式人生 > 實用技巧 >LibreOJ - 6277 數列分塊入門 1(分塊模板)

LibreOJ - 6277 數列分塊入門 1(分塊模板)

給出一個長為 nn 的數列,以及 nn 個操作,操作涉及區間加法,單點查值。

Input

第一行輸入一個數字 nn。

第二行輸入 nn 個數字,第 ii 個數字為 aiai,以空格隔開。

接下來輸入 nn 行詢問,每行輸入四個數字 optopt、ll、rr、cc,以空格隔開。

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

opt=1opt=1,表示詢問 arar 的值(ll 和 cc 忽略)。

Output

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

Example

樣例輸入

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

樣例輸出

2
5
#include <bits/stdc++.h>
using namespace std;
long long a[50005], sum[50005],add[50005];
int L[50005], R[50005];
int pos[50005];
int n, t;
void change(int l, int r, long long d)
{
    int p = pos[l], q = pos[r];
    if(p == q)
    {
        for(int i = l; i <=r; i++) a[i] += d;
        sum[p] += d * (r - l + 1
); } else { for(int i = p + 1; i <= q - 1; i++) add[i] += d; for(int i = l; i <= R[p]; i++) a[i] += d; sum[p] += d * (R[p] - l + 1); for(int i = L[q]; i <= r; i++) a[i] += d; sum[q] += d * (r - L[q] + 1); } } long long ask(int l, int r) {
int p = pos[l], q = pos[r]; long long ans = 0; if(p == q) { for(int i = l; i <= r; i++) ans += a[i]; ans += add[p] * (r - l + 1); } else { for(int i = p + 1; i <= q - 1; i++) ans += sum[i] + add[i] * (R[i] - L[i] + 1); for(int i = l; i <= R[p]; i++) ans += a[i]; ans += add[p] * (R[p] - l + 1); for(int i = L[q]; i <= r; i++) ans += a[i]; ans += add[q] * (r - L[q] + 1); } return ans; } int main() { cin >> n; for(int i = 1; i <= n; i++) { scanf("%lld", &a[i]); } t = sqrt(n); for(int i = 1; i <= t; i++) { L[i] = (i - 1) * sqrt(n) + 1; R[i] = i * sqrt(n); } if(R[t] < n) t++, L[t] = R[t - 1] + 1,R[t] = n; for(int i = 1; i <= t; i++)//每一塊 { for(int j = L[i]; j <= R[i]; j++) { pos[j] = i; sum[i] += a[j]; } } for(int i = 1; i <= n; i++) { int opt, l, r; long long c; scanf("%d%d%d%lld", &opt, &l, &r, &c); if(opt == 0) { change(l, r, c); } else { cout << ask(r, r) << endl; } } }