Cod sursa(job #2064472)

Utilizator leraValeria lera Data 12 noiembrie 2017 13:37:24
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.28 kb
#include <iostream>
#include <fstream>
#define Nmax 15005
using namespace std;

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

int n, m, arb[4 * Nmax], sum, a[Nmax], op, x, y;
void upd(int nod, int poz, int val, int L, int R)
{
    int mij = (L + R) / 2;
    if(L == R)
    {
        arb[nod] = arb[nod] + val;
        return;
    }
    if(poz <= mij)
        upd(2 * nod, poz, val, L, mij);
    else upd(2 * nod + 1, poz, val, mij + 1, R);
    arb[nod] = arb[2 * nod] + arb[2 * nod + 1];
}
void query(int nod, int l, int r, int L, int R)
{
    int mij = (L + R)/2;
    if(l <= L && R <= r)
    {
        sum += arb[nod];
        return;
    }
    if(l > R || r < L)return;
    query(2 * nod, l, r, L, mij);
    query(2 * nod + 1, l, r, mij + 1, R);
}
int main()
{
    fin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        fin >> a[i];
        upd(1, i, a[i], 1, n);
    }
    for(int i = 1; i <= m; i++)
    {
        fin >> op >> x >> y;
        if(op == 0)//s - a achitat valoarea y din ziua x
        {
            upd(1, x, -y, 1, n);
        }
        else // suma sumelor restante a zilelor x-y
        {
            sum = 0;
            query(1, x, y, 1, n);
            fout << sum << '\n';
        }
    }
    return 0;
}