Cod sursa(job #2502321)

Utilizator hurjui12AlexandruHurjui Alexandru-Mihai hurjui12Alexandru Data 30 noiembrie 2019 17:50:57
Problema Datorii Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.15 kb
#include <fstream>
using namespace std;

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

int arb[1<<15];
int a, b;

void update (int nod, int st, int dr)
{
    if (st == dr)
        arb[nod] = b;
    else
    {
        int mij = (st+dr)>>1;
        if (a <= mij)
            update (nod<<1, st, mij);
        else
            update (nod<<1|1, mij+1, dr);
        arb[nod] = arb[nod<<1] + arb[nod<<1|1];
    }
}

int query(int nod, int st, int dr)
{
    if (a <= st && dr <= b)
        return arb[nod];
    else
    {
        int eins = 0, zwei = 0, mij = (st+dr)>>1;
        if (a <= mij)
            eins = query(nod<<1, st, mij);
        if (mij < b)
            zwei = query(nod<<1|1, mij+1, dr);
        return eins + zwei;
    }
}

int main()
{
    int n, i, m, v[15001], t;
    fin >> n >> m;
    for (a = 1; a<=n; a++)
    {
        fin >> v[a];
        b = v[a];
        update(1, 1, n);
    }
    for (i = 1; i<=m; i++)
    {
        fin >> t >> a >> b;
        if (t == 0)
        {
            v[a] = v[a] - b;
            b = v[a];
            update(1, 1, n);
        }
        else
            fout << query(1, 1, n) << '\n';
    }
    return 0;
}