Cod sursa(job #2902256)

Utilizator ralucarRogoza Raluca ralucar Data 15 mai 2022 23:02:22
Problema Datorii Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.72 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int n, m, arbint[1000001], operatie, x, a, b;
int suma(int nod, int stanga, int dreapta, int pozitie_stanga, int pozitie_dreapta)
{
    if(pozitie_stanga>dreapta || pozitie_dreapta<stanga)
        return 0;
    if(stanga>=pozitie_stanga && dreapta<=pozitie_dreapta)
        return arbint[nod];
    int mijloc=(stanga+dreapta)/2;
    return suma(nod*2, stanga, mijloc, pozitie_stanga, pozitie_dreapta)+suma(nod*2+1, mijloc+1, dreapta, pozitie_stanga, pozitie_dreapta);
}

void creare(int nod, int stanga, int dreapta, int pozitie, int valoare)
{
    if(stanga==dreapta)
    {
        arbint[nod]=valoare;
        return;
    }
    int mijloc=(stanga+dreapta)/2;
    if(mijloc>=pozitie)
        creare(nod*2, stanga, mijloc, pozitie, valoare);
    else
        creare(nod*2+1, mijloc+1, dreapta, pozitie, valoare);
    arbint[nod]=arbint[nod*2]+arbint[nod*2+1];
}

void update(int nod, int stanga, int dreapta, int pozitie, int valoare)
{
    if(stanga==dreapta)
    {
        arbint[nod]-=valoare;
        return;
    }
    int mijloc=(stanga+dreapta)/2;
    if(mijloc>=pozitie)
        update(nod*2, stanga, mijloc, pozitie, valoare);
    else
        update(nod*2+1, mijloc+1, dreapta, pozitie, valoare);
    arbint[nod]=arbint[nod*2]+arbint[nod*2+1];
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=n; i++)
    {
        fin>>x;
        creare(1, 1, n, i, x);
    }

    for(int i=1; i<=m; i++)
    {
        fin>>operatie>>a>>b;
        if(operatie==0)
            update(1, 1, n, a, b);
        else
            fout<<suma(1, 1, n, a, b)<<'\n';

    }
    return 0;
}