Cod sursa(job #3267125)

Utilizator Tud_17Rednic Tudor Tud_17 Data 11 ianuarie 2025 09:51:19
Problema Arbori indexati binar Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.41 kb
#include <iostream>
using namespace std;


int n, m, c, x;
int a, b;
int pos;
int aib[100005]; // dimensiune N


void update(int i, int x) {
    for (int j=i;j<=n;j+=(j&-j)) { // j+=(j&-j) -> trecem prin numerele care il contin in suma pe elementul de pe pozitia i
        aib[j] += x;
    }
}


void create() {
    for (int i=1;i<=n;i++) {
        cin >> x;
        update(i, x);
    }
}


int query(int i) {
    int s = 0;
    for (int j=i;j>=1;j-=(j&-j)) { // j -= (j&-j) -> eliminam cel mai din dreapta bit de 1 la fiecare pas, j&(-j) reprezinta cel mai din dreapta bit de 1, putem sa il eliminam deoarece j contine exact suma a j&(-j) elemente, deci cu exact atatea elemente mergem in spate
        s += aib[j];
    }
    return s;
}

int solve2()
{
    int s;
    for(int i = 1; i <= n; i++){
        s = query(i);
        if(s == a)
            return i;
        if(s > a)
            return -1;
    }
    return -1;
}

void solve() {
    cin >> c;
    if (c == 0) {
        cin >> a >> x;
        update(a, x);
    }
    else if (c == 1) {
        cin >> a >> b;
        cout << query(b) - query(a-1) << '\n';
    }
    else {
        cin >> a;
        cout << solve2() << '\n';
    }
}


int main() {
    freopen("aib.in", "r", stdin);
    freopen("aib.out", "w", stdout);
    cin >> n >> m;
    create();
    for(;m;m--) {
        solve();
    }
    return 0;
}