LibreOJ - 6277 数列分块入门 1（分块模板）

Posted 2020-12-17 lipoicyclic

给出一个长为 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;
        }
    }
    
}