Cod sursa(job #2750636)

Utilizator Andrei_TudorAndrei Tudor Andrei_Tudor Data 12 mai 2021 17:19:43
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.45 kb
#include <fstream>

using namespace std;
const int NMAX = 100000;
int aib[NMAX + 5];
int n;
void update(int poz, int val){
    for(; poz <= n; poz += (poz ^ (poz & (poz - 1)))){
        aib[poz] += val;
    }
}
int query(int poz){
    int sum = 0;
    for(; poz >= 1; poz -= (poz ^ (poz & (poz - 1)))){
        sum += aib[poz];
    }
    return sum;
}
ifstream cin("aib.in");
ofstream cout("aib.out");
int main()
{
    int m, i, a, b, op, last;
    cin >> n >> m;
    for(i = 1; i <= n; i ++){
        cin >> a;
        update(i, a);
    }
    for(i = 1; i <= m; i ++){
        cin >> op >> a;
        if(op == 0){
            cin >> b;
            update(a, b);
        }
        else if(op == 1){
            cin >> b;
            cout << query(b) - query(a - 1) << "\n";
        }
        else if(op == 2){
            int st = 1;
            int dr = n;
            int med;
            last = -1;
            while(st <= dr){
                med = (st + dr) / 2;
                int x = query(med);
                if(x < a){
                    st = med + 1;
                }
                else {
                    if(x >= a){
                        last = med;
                        dr = med - 1;
                    }
                }
            }
            if(last != -1 && query(last) != a){
                last = -1;
            }
            cout << last << "\n";
        }
    }
    return 0;
}