码迷,mamicode.com
首页 > 其他好文 > 详细

LibreOJ - 6277 数列分块入门 1(分块模板)

时间:2020-07-12 13:57:38      阅读:38      评论:0      收藏:0      [点我收藏+]

标签:data   col   code   out   输入   -o   etc   div   ++   

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

 

LibreOJ - 6277 数列分块入门 1(分块模板)

标签:data   col   code   out   输入   -o   etc   div   ++   

原文地址:https://www.cnblogs.com/lipoicyclic/p/13288036.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!