243. 一个简单的整数问题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|≤100000,
|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

思路：

用树状数组完成 O(logn)
时间复杂度的区间查询 + 区间修改

因此只需维护两个树状数组即可
一个是差分数组d[i]的树状数组tr[i]，还有一个是i*d[i]的树状数组tri[i]

代码：

#include<bits/stdc++.h>
using namespace std;

#define x first
#define y second
#define endl '\n'
#define IOS ios::sync_with_stdio(false); cin.tie(0);cout.tie(0);
typedef long long LL;
typedef unsigned long long ull;
typedef pair<int, int> PII;

LL lowbit(LL x) { return x & (- x); }
LL gcd(LL a, LL b) { return b ? gcd(b, a % b) : a; }
const int mod = 1e9 + 7;
const int N = 100010;

LL n, m;
LL a[N];
LL tr1[N];  // 维护差分数组b[i]的前缀和
LL tr2[N];  // 维护b[i] * i的前缀和

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

LL sum (LL tr[], int x)
{
    LL res = 0;

    for (int i = x; i; i -= lowbit(i))
        res += tr[i];

    return res;
}


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];

    for (int i = 1 ; i <= n; i++)
    {
        LL b = a[i] - a[i - 1];
        add(tr1, i, b);
        add(tr2, i, b * i);
    }

    while (m--)
    {
        char op[2];
        int l, r, d;
        cin >> op >> l >> r;

        if (op[0] == 'Q')
        {
            cout << prefix_sum(r) - prefix_sum(l - 1) << endl;
        }
        else
        {
            cin >> d;
            add(tr1, l, d), add(tr2, l, l * d);
            add(tr1, r + 1, -d), add(tr2, r + 1, (r + 1) * (-d));
        }
    }

    return 0;
}