Cod sursa(job #2427523)

Utilizator PrekzursilAndrei Visalon Prekzursil Data 31 mai 2019 23:08:27
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.11 kb
#include <fstream>
using namespace std;
ifstream in("aib.in");
ofstream out("aib.out");
int aib[100010], n, m;
int query (int p)
{
    int s=0;
    while (p>0)
    {
        s+=aib[p];
        p-=p&(-p);
    }
    return s;
}
void update (int p, int val)
{
    while (p<=n)
    {
        aib[p]+=val;
        p+=p&(-p);
    }
}
int caut(int val)
{
    int r=0, pas=1<<16;
    if (query(n) < val) return -1;
    while (pas != 0)
    {
        if (r + pas <= n && query(r + pas) < val)
            r+=pas;
        pas/=2;
    }
    if (query(r+1) != val) r=-1;
    else r++;
    return r;
}
int main()
{
    int a, b, nr;
    in>>n>>m;
    for (int i=1; i<=n; i++)
    {
        in>>nr;
        update (i, nr);
    }
    for (int i=1; i<=m; i++)
    {
        in>>nr;
        if (nr==0)
        {
            in>>a>>b;
            update (a, b);
        }
        if (nr==1)
        {
            in>>a>>b;
            out<<query(b)-query(a-1)<<"\n";
        }
        if (nr==2)
        {
            in>>a;
            out<<caut(a)<<"\n";
        }
    }
    return 0;
}