一个简单的整数问题2

题目:

题目链接

题解:

树状数组

#include <bits/stdc++.h>
using namespace std;
long long a[100005],t1[100005],t2[100005],sum[100005];
int n;
void add1(int x, int k)
{
    for(; x <= n; x += x & -x) t1[x] += k;
}
long long ask1(int x)
{
    long long ans = 0;
    for(; x; x -= x & -x) ans += t1[x];
    return ans;
}
void add2(int x, int k)
{
    for(; x <= n; x += x & -x) t2[x] += k;
}
long long ask2(int x)
{
    long long ans = 0;
    for(; x; x -= x & -x) ans += t2[x];
    return ans;
}
int main()
{
    int m;
    cin>>n>>m;
    for(int i = 1; i <= n; i ++ ) cin>>a[i],sum[i] = sum[i - 1] + a[i];
    while(m -- )
    {
        char op;
        cin>>op;
        if(op == 'C')
        {
            int L, R, d;
            cin>>L>>R>>d;
            add1(L, d);
            add1(R + 1, -d);
            add2(L, L * d);
            add2(R + 1, -(R + 1) * d);
        }
        else 
        {
            int L, R;
            cin>>L>>R;
            long long ans = (sum[R] + (R + 1) * ask1(R) - ask2(R)) - (sum[L - 1] + L * ask1(L - 1) - ask2(L - 1));
            cout<<ans<<endl;
        }
    }
    return 0;
}