Cod sursa(job #2707838)

Utilizator butasebiButa Gabriel-Sebastian butasebi Data 17 februarie 2021 19:51:33
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.31 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 f("aib.in");
    ofstream g("aib.out");
    f >> n >> q;
    for(i = 1; i <= n; i ++)
    {
        f >> x;
        add_aib(i, x);
    }
    for(i = 1; i <= q; i ++)
    {
        f >> op;
        if(op == 0)
        {
            f >> a >> b;
            add_aib(a, b);
        }
        if(op == 1)
        {
            f >> a >> b;
            g << sum_aib(b) - sum_aib(a - 1) << "\n";
        }
        if(op == 2)
        {
            f >> 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;
            }
            g << p << "\n";
        }
    }
    return 0;
}