Cod sursa(job #2615812)

Utilizator DeliaGhergheGherghe Ioana-Delia DeliaGherghe Data 15 mai 2020 17:06:46
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.54 kb
#include <iostream>
#include <fstream>
using namespace std;

int arbint[60001], A[15001], N, M;

void build(int nod, int st, int dr)
{
    if (st == dr)
    {
        arbint[nod] = A[st];
        return;
    }
    else
    {
        int mij = (st + dr)/2;
        build(nod*2, st, mij);
        build(nod*2 + 1, mij + 1, dr);
        arbint[nod] = arbint[nod*2] + arbint[nod*2 +1];
    }
}

void update(int nod, int st, int dr, int poz, int s)
{
    if (st == dr)
    {
        arbint[nod] = arbint[nod] - s;
        return;
    }
    else
    {
        int mij = (st + dr)/2;
        if (poz <= mij)
            update(nod*2, st, mij, poz, s);
        else update(nod*2+1, mij+1, dr, poz, s);
        arbint[nod] = arbint[nod*2] + arbint[nod*2+1];
    }
}

int query(int nod, int st, int dr, int sti, int dri)
{
    if (sti <= st && dr <= dri)
        return arbint[nod];
    int s = 0, mij = (st + dr)/2;
    if (sti <= mij)
        s = s + query(nod*2, st, mij, sti, dri);
    if (dri > mij)
        s = s + query(nod*2 + 1, mij + 1, dr, sti, dri);
    return s;
}

int main()
{
    ifstream fin("datorii.in");
    ofstream fout("datorii.out");

    int i,x,y,z;
    fin >> N >> M;

    for (i = 1; i<= N; i++)
        fin >> A[i];
    build(1,1,N);

    for (i = 1; i <= M; i++)
    {
        fin >> x >> y >> z;
        if (x == 0)
            update(1,1,N,y,z);
        else if (x == 1)
            fout << query(1,1, N,y,z) << '\n';
    }
    fin.close();
    fout.close();
    return 0;
}