AcWing 243 一个简单的整数问题2

AcWing 243

题目描述

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

C l r d，表示把 A[l],A[l+1],…,A[r] 都加上 d。
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 <iostream>
using namespace std;

typedef long long LL;

int N,M;
int a[100001];
LL tr[100001],tri[100001];
int lowbit(int x)
{
    return x & -x;
}
void update(int x, LL c, LL tr[])  // 位置x加c
{
    for (int i = x; i <= N; i += lowbit(i)) tr[i] += c;
}
LL query(int x, LL tr[])  // 返回前x个数的和
{
    LL res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}
LL getsum(int x)
{
    return query(x,tr)*(x+1)-query(x,tri);
}
int main()
{
    cin>>N>>M;
    for(int i=1;i<=N;i++) cin>>a[i];
    for(int i=1;i<=N;i++){
        update(i,a[i]-a[i-1],tr);
        update(i,1ll*(a[i]-a[i-1])*i,tri);
    }
    for(int i=1;i<=M;i++){
        string op;
        cin>>op;
        if(op=="Q"){
            int l,r;
            cin>>l>>r;
            cout<<getsum(r)-getsum(l-1)<<endl;
        }else{
            int l,r,d;
            cin>>l>>r>>d;
            update(l,d,tr); update(r+1,-d,tr);
            update(l,d*l,tri); update(r+1,-d*(r+1),tri);
        }
    }
    
    return 0;
}