Cod sursa(job #2485333)

Utilizator theo2003Theodor Negrescu theo2003 Data 1 noiembrie 2019 12:42:39
Problema Datorii Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.48 kb
#include <fstream>
using namespace std;
ifstream cin("datorii.in");
ofstream cout("datorii.out");
int l[15001];
struct aint{
    aint *a = NULL, *b = NULL;
    int l, r, v = 0;
    aint(int x, int y){
        l = x;
        r = y;
    }
};
int build(aint *node){
    if(node->l == node->r){
        //cout<<l[node->l]<<"!\n";
        return node->v = l[node->l];
    }else{
        int m = node->l + (node->r - node->l)/2;
        node->a = new aint(node->l, m);
        node->b = new aint(m + 1, node->r);
        return node->v = build(node->a) + build(node->b);
    }
}
void add(aint *node, int target, int val){
    node->v += val;
    if(node->l != node->r){
        if(target <= node->a->r)
            add(node->a, target, val);
        else add(node->b, target, val);
    }
}
int query(aint *node, int l, int r){
    if((l == node->l) && (r == node->r))
        return node->v;
    else if(r <= node->a->r)
        return query(node->a, l, r);
    else if(l >= node->b->l)
        return query(node->b, l, r);
    else return query(node->a, l, node->a->r) + query(node->b, node->b->l, r);
}
int main(){
    int n, m;
    cin>>n>>m;
    for(int x = 1;x<=n;x++)
        cin>>l[x];
    aint *root = new aint(1, n);
    //cout<<
    build(root);
    for(int x = 0;x<m;x++){
        int a, b, c;
        cin>>a>>b>>c;
        if(a){
            cout<<query(root, b, c)<<'\n';
        }else{
            add(root, b, -c);
        }
    }
    return 0;
}