Cod sursa(job #2781170)

Utilizator florinrafiliuRafiliu Florin florinrafiliu Data 8 octombrie 2021 18:07:08
Problema Datorii Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.61 kb
#include <iostream>
#include <fstream>
using namespace std;

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

int tree[60005];
int v[15005];

void build (int node, int st, int dr) {
    if(st == dr)
        tree[node] = v[st];
    else {
        int mij = (st + dr) / 2;
        build(node*2, st, mij);
        build(node*2+1, mij+1, dr);
        tree[node] = tree[node*2] + tree[node*2+1];
    }
}

void update (int node, int st, int dr, int ind, int new_val) {
    if(st == dr) {
        tree[node] = tree[node] - new_val;
    } else {
        int mij = (st + dr) / 2;
        if(ind <= mij)
            update(node*2, st, mij, ind, new_val);
        if(ind > mij)
            update(node*2+1, mij+1, dr, ind, new_val);
        tree[node] = tree[node*2] + tree[node*2+1];
    }
}

int suma (int node, int st, int dr, int le, int ri) {
    if(st > dr) return 0;
    if(st == le && dr == ri)
        return tree[node];
    else {
        int mij = (st + dr) / 2;
        int suma1 = 0, suma2 = 0;
        if(le <= mij)
            suma1 = suma(node*2, st, mij, le, min(mij, ri));
        if(ri > mij)
            suma2 = suma(node*2+1, mij+1, dr, max(le, mij+1), ri);

        return suma1 + suma2;
    }
}

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

    build(1, 1, n);

    for(int i = 1; i <= m; ++i){
        int op, x, y;
        fin >> op >> x >> y;
        if(op == 1) {
            fout << suma(1, 1, n, x, y) << "\n";
        } else {
            update(1, 1, n, x, y);
        }
    }

    return 0;
}