Cod sursa(job #2064838)

Utilizator CammieCamelia Lazar Cammie Data 12 noiembrie 2017 21:44:32
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.55 kb
#include <fstream>

#define MAXN 250004

using namespace std;

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

int ait[MAXN * 3], in[MAXN];

inline void build(int poz, int st, int dr) {
    int mij = (st + dr) / 2;

    if (st == dr)
        ait[poz] = in[st];
    else {
        build(poz * 2, st, mij);
        build(poz * 2 + 1, mij + 1, dr);

        ait[poz] = ait[poz * 2] + ait[poz * 2 + 1];
    }
}

inline int Query(int poz, int st, int dr, int ls, int ld) {
    int mij = (st + dr) / 2;

    if (ld < st || ls > dr)
        return 0;
    if (st >= ls && dr <= ld)
        return ait[poz];
    else {
        return Query(poz * 2, st, mij, ls, ld) + Query(poz * 2 + 1, mij + 1, dr, ls, ld);
    }
}

inline void Update(int poz, int st, int dr, int a, int val) {
    int mij = (st + dr) / 2;

    if (st == dr && st == a) {
        ait[poz] -= val;
        return;
    }
    if (st > a || dr < a)
        return;
    Update(poz * 2, st, mij, a, val);
    Update(poz * 2 + 1, mij + 1, dr, a, val);
    ait[poz] = ait[poz * 2] + ait[poz * 2 + 1];
}

inline void Read() {
    int N, M, tip, a, b;

    fin >> N >> M;

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

    build(1, 1, N);

    for (int i = 1; i <= M; i++) {
        fin >> tip >> a >> b;

        if (tip == 1) {
            fout << Query(1, 1, N, a, b) << "\n";
        }
        else {
            Update(1, 1, N, a, b);
        }
    }
}

int main () {
    Read();

    fin.close(); fout.close(); return 0;
}