一个简单的整数问题2 - 区间修改+区间查询

给定一个长度为 N 的数列 A，以及 M 条指令，每条指令可能是以下两种之一：

1. C l r d，表示把 A[l],A[l+1],…,A[r] 都加上 d。
2. Q l r，表示询问数列中第 l∼r 个数的和。

对于每个询问，输出一个整数表示答案。

输入格式

第一行两个整数 N,M。

第二行 N 个整数 A[i]。

接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

1≤N,M≤105,
|d|≤10000,
|A[i]|≤109

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

输出样例：

4
55
9
15

 

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'

using namespace std;

typedef pair<int, int> PII;
typedef long long ll;

const int N = 100010;

int n, m;
int a[N];
ll tr1[N];//b[i]
ll tr2[N];//b[i] * i b*i可能会爆ll，写函数的时候注意一下。

inline int lowbit(int x)
{
    return x & -x;
}

inline void add(ll tr[], int x, ll c)
{
    for(int i = x; i <= n; i += lowbit(i))
    {
        tr[i] += c;
    }
}

inline ll sum(ll tr[], int x)
{
    ll res = 0;
    for(int i = x; i; i -= lowbit(i))
    {
        res += tr[i];
    }
    return res;
}

inline ll prefix_sum(int x)
{
    return sum(tr1, x) * (x + 1) - sum(tr2, x);
}

int main()
{
    IOS
    cin >> n >> m;
    for(int i = 1; i <= n; i ++)
    {
        cin >> a[i];
        int b = a[i] - a[i - 1];
        add(tr1, i, b), add(tr2, i, (ll)b * i);
    }
    
    char op[2];
    int l, r, d;
    while(m --)
    {
        cin >> op;
        if(op[0] == 'C')
        {
            cin >> l >> r >> d;
            add(tr1, l, d), add(tr1, r + 1, -d);
            add(tr2, l, d * l), add(tr2, r + 1, (r + 1) * -d);
        }
        else
        {
            cin >> l >> r;
            cout << prefix_sum(r) - prefix_sum(l - 1) << endl;
        }
    }
    
    return 0;
}

差分数组的前缀和是单点查询，差分数组前缀和的前缀和就可以用来区间查询。

 a1到ax的前缀和可以看作：

学自：Acwing