Cod sursa(job #3294497)

Utilizator stefan_dore_Stefan Dore stefan_dore_ Data 24 aprilie 2025 16:49:33
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.2 kb
#include <iostream>
#include <fstream>
using namespace std;

ifstream f ("datorii.in");
ofstream g ("datorii.out");

const int NMAX = 15000;
int N, M, Ai[4*NMAX+1];

void Update(int rad, int st, int dr, int poz, int val) {
    if (st == dr)
        Ai[rad] -= val;
    else {
        int m = (st + dr) / 2;
        //
        if (poz <= m) Update(2*rad, st, m, poz, val);
        else Update(2*rad+1, m+1, dr, poz, val);
        //
        Ai[rad] = Ai[2*rad] + Ai[2*rad+1];
    }
}

int Query(int rad, int st, int dr, int a, int b) {
    if (a <= st && b >= dr)
        return Ai[rad];
    else {
        int m = (st + dr) / 2,
            r1 = 0, r2 = 0;
        //
        if (a <= m) r1 = Query(2*rad, st, m, a, b);
        if (b > m) r2 = Query(2*rad+1, m+1, dr, a, b);
        //
        return r1 + r2;
    }
}

int main(){
    int x;
    f >> N >> M;
    for (int i=1; i<=N; i++) {
        f >> x;
        Update(1, 1, N, i, -x);
    }
    //
    int op, p, q;
    while(M--) {
        f >> op >> p >> q;
        //
        if (op == 0) Update(1, 1, N, p, q);
        else g << Query(1, 1, N, p, q) << '\n';
    }
    //
    f.close();
    g.close();
    return 0;
}