Cod sursa(job #1843526)

Utilizator stefanmereutaStefan Mereuta stefanmereuta Data 8 ianuarie 2017 20:42:33
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.95 kb
#include <iostream>
#include <fstream>

using namespace std;

#define MAX 15010

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

        carb(v, arb, 2 * nod, st, mid);
        carb(v, arb, 2 * nod + 1, mid + 1, dr);

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

void update(int arb[]/*, int n*/, int poz, int val, int nod, int st, int dr) {
    //int st = 1, dr = n, mid, nod = 1;
    if (st == dr) {
        arb[poz] -= val;
        return;
    }

    int mid = (st + dr) / 2;

    if (poz <= mid) {
        update(arb, poz, val, nod * 2, st, mid);
    } else {
        update(arb, poz, val, nod * 2 + 1, mid + 1, dr);
    }

    arb[nod] -= val;
/*
    while (st < dr) {
        mid = (st + dr) / 2;

        if (mid < poz) {
            nod = 2 * nod + 1;
            st = mid + 1;
        } else {
            nod = 2 * nod;
            dr = mid;
        }
    }

    while (nod > 0) {
        arb[nod] -= val;
        nod /= 2;
    }*/
}

int query(int arb[], int st_q, int dr_q, int nod, int st, int dr) {
    if (st_q <= st && dr <= dr_q) {
        return arb[nod];
    }

    if (dr < st_q || dr_q < st) {
        return 0;
    }

    int mid = (st + dr) / 2;

    return query(arb, st_q, dr_q, 2 * nod, st, mid) + query(arb, st_q, dr_q, 2 * nod + 1, mid + 1, dr);
}

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

    int n, m, v[MAX], arb[4 * MAX], op, a, b;

    fin >> n >> m;

    for (int i = 1; i <= n; i++) {
        fin >> v[i];
    }

    carb(v, arb, 1, 1, n);

    for (int i = 0; i < m; i++) {
        fin >> op >> a >> b;

        if (op == 0) {
            update(arb, a, b, 1, 1, n);
        } else {
            fout << query(arb, a, b, 1, 1, n) << endl;
        }
    }

    fin.close();
    fout.close();

    return 0;
}