Cod sursa(job #3213161)

Utilizator camelia22Dragoiu Camelia camelia22 Data 12 martie 2024 16:46:25
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.56 kb
#include <iostream>
#include <fstream>
#define N 15005
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int arb[5*N], v[N], n, m, a, b, i, cod;

void construire(int nod, int st, int dr)
{
    if (st == dr)
        arb[nod] = v[st];
    else
    {
        int mij = (st + dr) / 2;

        construire(nod * 2, st, mij);
        construire(nod * 2 + 1, mij + 1, dr);

        arb[nod] = arb[nod * 2] + arb[nod * 2 + 1];
    }
}

void actualizare(int nod, int st, int dr, int zi, int val)
{
    if (st == dr)
        arb[nod] -= val;
    else
    {
        int mij = (st + dr) / 2;

        if (zi <= mij)
            actualizare(nod * 2, st, mij, zi, val);
        else
            actualizare(nod * 2 + 1, mij + 1, dr, zi, val);

        arb[nod] = arb[nod * 2] + arb[nod * 2 + 1];
    }
}

int suma(int nod, int st, int dr, int a, int b)///intervalul [a,b]
{
    if (a <= st && b >= dr)
        return arb[nod];
    else
    {
        int mij = (st + dr) / 2;

        if (b <= mij)
            return suma(nod * 2, st, mij, a, b);
        if (a >= mij + 1)
            return suma(nod * 2 + 1, mij + 1, dr, a, b);

        return suma(nod * 2, st, mij, a, b) + suma(nod * 2 + 1, mij + 1, dr, a, b);
    }
}

int main()
{
    f >> n >> m;
    for (i = 1; i <= n; i++)
        f >> v[i];
    construire(1,1,n);

    while (m--)
    {
        f >> cod >> a >> b;
        if (cod == 0)
            actualizare(1,1,n,a,b);
        else
            g << suma(1,1,n,a,b) << '\n';
    }
    return 0;
}