Cod sursa(job #2903334)

Utilizator deboradeleanuDebora Deleanu deboradeleanu Data 17 mai 2022 14:12:27
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.18 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in("datorii.in");
ofstream out("datorii.out");
int arb[40005], v[15001];

int suma(int n, int st, int dr, int x, int y)
{
    if(dr < x || st > y)
        return 0;
    if(x <= st && dr <= y)
        return arb[n];
    return suma(n*2, st, (st + dr)/2, x, y) + suma(n*2+1, (st + dr)/2+1, dr, x, y);
}

void creeaza(int n, int st, int dr)
{
    if(st == dr)
    {
        arb[n] = v[st];
        return;
    }
    creeaza(n*2, st, (st + dr)/2);
    creeaza(n*2+1, (st + dr)/2 + 1, dr);
    arb[n] = arb[n*2] + arb[n*2+1];
}
void f(int n, int st, int dr, int poz, int x)
{
    if(poz < st || poz > dr)return;
    if(st == dr)
    {
        arb[n] -= x;
        return;
    }
    f(2*n, st, (st + dr)/2, poz, x);
    f(2*n + 1, (st + dr)/2 + 1, dr, poz, x);
    arb[n] = arb[2*n] + arb[2*n+1];
}
int main()
{
    int n, m;
    in>>n>>m;
    for(int i = 1; i <= n; i++)
        in>>v[i];
    creeaza(1, 1, n);
    int op, x, y;
    for(int i = 0; i < m; i++)
    {
        in>>op>>x>>y;
        if(op)
            out<<suma(1, 1, n, x, y)<<"\n";
        else f(1, 1, n, x, y);
    }
    return 0;
}