Cod sursa(job #2802764)

Utilizator AndreiCroitoruAndrei Croitoru AndreiCroitoru Data 18 noiembrie 2021 19:19:20
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.56 kb
#include <bits/stdc++.h>

using namespace std;

int i, x, q, a, b, n, st, dr, mij, p, op;
long long aux, aib[100005];

int LastBit(int x)
{
    return x & (-x);
}
void add_aib(int poz, int val)
{
    for(int i = poz; i <= n; i = i + LastBit(i))
    {
        aib[i] = aib[i] + val;
    }
}
long long sum_aib(int poz)
{
    long long sum = 0;
    for(int i = poz; i > 0; i = i - LastBit(i))
    {
        sum = sum + aib[i];
    }
    return sum;
}
int main()
{
    ifstream cin("aib.in");
    ofstream cout("aib.out");
    cin >> n >> q;
    for(i = 1; i <= n; i ++)
    {
        cin >> x;
        add_aib(i, x);
    }
    for(i = 1; i <= q; i ++)
    {
        cin >> op;
        if(op == 0)
        {
            cin >> a >> b;
            add_aib(a, b);
        }
        if(op == 1)
        {
            cin >> a >> b;
            cout << sum_aib(b) - sum_aib(a - 1) << "\n";
        }
        if(op == 2)
        {
            cin >> a;
            st = 1;
            dr = n;
            p = -1;
            while(st <= dr)
            {
                mij = (st + dr) / 2;
                aux = sum_aib(mij);
                if(aux == a)
                {
                    p = mij;
                    st = dr + 1;
                }
                else if(aux < a)
                {
                    st = mij + 1;
                }
                else
                {
                    dr = mij - 1;
                }
            }
            cout << p << "\n";
        }
    }
    return 0;
}