Cod sursa(job #2482866)

Utilizator theo2003Theodor Negrescu theo2003 Data 28 octombrie 2019 22:53:03
Problema Arbori indexati binar Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.16 kb
#include <fstream>
//#include <iostream>
using namespace std;
ifstream cin("aib.in");
ofstream cout("aib.out");
struct aint {
    aint *a, *b;
    int l, r;
    int v;
    aint(int y, int z) {
        l = y;
        r = z;
        v = 0;
        a = NULL;
        b = NULL;
    }
    aint(){}
};
void build(aint *node) {
    int l = node->l;
    int r = node->r;
    if(l != r) {
        node->a = new aint(l, l + (r - l + 1)/2 - 1);
        node->b = new aint(l + (r - l + 1)/2, r);
        build(node->a);
        build(node->b);
    }
}
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);
}
void add(aint *node, int add, int target){
    while(true){
        node->v += add;
        if(node->a == NULL)
            return;
        if(node->a->r >= target)
            node = node->a;
        else node = node->b;
    }
}
int main() {
    int n, m;
    cin>>n>>m;
    aint root = aint(1, n);
    build(&root);
    for(int x = 1, tmp; x<=n; x++){
        cin>>tmp;
        add(&root, tmp, x);
    }
    for(int x = 0, req;x<m;x++){
        cin>>req;
        if(req == 0){
            int target, add_;
            cin>>target>>add_;
            add(&root, add_, target);
        }else if(req == 1){
            int l, r;
            cin>>l>>r;
            cout<<query(&root, l, r)<<'\n';
        }else{
            int v;
            cin>>v;
            int l = 1, r = n, m;
            while(l != r){
                m = l + (r - l)/2 + 1;
                int v1 = query(&root, 1, m);
                if(v1 == v){
                    l = m;
                    r = m;
                    break;
                }else if(v1 < v){
                    l = m + 1;
                }else r = m - 1;
            }
            m = l;
            if(query(&root, 1, m) != v)
                cout<<-1<<'\n';
            else cout<<l<<'\n';
        }
    }
    return 0;
}