Cod sursa(job #2712028)

Utilizator Ricardo03Petrovici Ricardo Ricardo03 Data 25 februarie 2021 08:35:27
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <bits/stdc++.h>
using namespace std;
int v[100005], sum, n, m, i, x, op, a, b, st, dr, mij;
int last_bit(int y)
{
    return y & (-y);
}
void add_aib(int poz, int val)
{
    for(int i = poz; i <= n; i = i + last_bit(i))
        v[i] = v[i] + val;
}
int sum_aib(int poz)
{
    int s = 0;
    for(int i = poz; i >= 1; i = i - last_bit(i))
        s = s + v[i];
    return s;
}
int main()
{
    ifstream f("aib.in");
    ofstream g("aib.out");
    f >> n >> m;
    for(i = 1; i <= n; i++)
    {
        f >> x;
        add_aib(i, x);
    }
    for(i = 1; i <= m; 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;
            while(st <= dr)
            {
                mij = (st + dr) / 2;
                sum = sum_aib(mij);
                if(sum == a)
                {
                    g << mij << "\n";
                    break;
                }
                else if(sum < a) st = mij + 1;
                else dr = mij - 1;
            }
        }
    }
    return 0;
}