Cod sursa(job #2567514)

Utilizator ikogamesIon Ceaun ikogames Data 3 martie 2020 17:41:04
Problema Arbori indexati binar Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.12 kb
#include <bits/stdc++.h>
#define zeros(x) (x&(-x))
using namespace std;
int AIB[100001];
int n, m;
ifstream fin ("aib.in");
ofstream fout ("aib.out");

void Add(int x, int quantity)
{
    for(int i = x; i <= n; i+=zeros(i))
        AIB[i] += quantity;
}

int Compute(int x)
{
    int ret = 0;
    for(int i = x; i >= 1; i -=zeros(i))
        ret += AIB[i];
    return ret;
}
void Read()
{
    int op, a, b;
    fin >> n >> m;
    for(int i = 1; i <= n; i++)
        {
            fin >> a;
            Add(i, a);
        }
    for(int i = 1; i <= m; i++)
    {
        fin >> op;
        if(op == 0)
        {
            fin >> a >> b;
            Add(a, b);

        }
        else if(op == 1)
        {
            fin >> a >> b;
            fout << Compute(b) - Compute(a - 1) << "\n";
        }
        else if(op == 2)
        {
            fin >> a;
            int poz;
            for(poz = 1; Compute(poz) != a && poz <= n; poz++)
                ;
            if(poz > n) poz = -1;
            fout << poz << "\n";
        }
    }
}

int main()
{
    Read();
    return 0;
}