AcWing 243. Un problema entero simple 2 (modificación de intervalo de la matriz de árbol + consulta de intervalo)

AcWing243. Un problema de entero simple 2 ( modificación de intervalo de la matriz de árbol + consulta de intervalo )

${un}_1+{un}_2+{un}_3+un...{un}_x$ $==$ ${\sum }_{i=1}^x{\sum }_{j=1}^{yo}{segundo}_j$

Dada una secuencia A de instrucciones de longitud N y M, cada instrucción puede ser una de las dos siguientes:

1. "C lrd" significa sumar d a todo A [l], A [l + 1], ..., A [r].

2. "Q lr" significa la suma del número l ~ r en la secuencia de consulta.

Para cada consulta, envíe un número entero para indicar la respuesta.

Formato de entrada

Dos números enteros N y M en la primera línea.

N números enteros A [i] en la segunda línea.

Las siguientes líneas M representan instrucciones M, y el formato de cada instrucción se muestra en la descripción del título.

Formato de salida

Para cada consulta, envíe un número entero para indicar la respuesta.

Cada respuesta está en una línea.

rango de datos

$1\le N,METRO\le 10^5,$
$|d|\le 10000$
$|A[i]|\le 1000000000$

Código

#include <iostream>

using namespace std;

typedef long long LL;

const int N = 100010;

int n , m;
int a[N];
LL tr1[N] , tr2[N];

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

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()
{
    
    
    scanf("%d%d" , &n , &m);

    for(int i = 1; i <= n ; i++)    scanf("%d", &a[i]);

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

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

        scanf("%s%d%d" , op , &l , &r);

        if(*op == 'Q')
            cout << prefix_sum(r) - prefix_sum(l - 1) << endl;
        else
        {
    
       
            scanf("%d" , &d);
            add(tr1 , l , d) , add(tr1 , r + 1 , -d);
            add(tr2 , l , l * d) , add(tr2 , r + 1 , (r + 1) * (- d));
        }
    }
        return 0;
}