Cod sursa(job #3333644)

Utilizator LucaWalucaLuca Munteanu LucaWaluca Data 14 ianuarie 2026 18:40:58
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.11 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("aib.in");
ofstream out("aib.out");
int aib[100005],n,v[100005];
void update(int poz, int nr)
{
    for(int i=poz;i<=n;i+=(i&-i))
        aib[i]+=nr;
}
int query(int poz)
{
    int sol=0;
    for(int i=poz;i>=1;i-=(i&-i))
        sol+=aib[i];
    return sol;
}
int main()
{
    int q,t,a,b;
    in>>n>>q;
    for(int i=1;i<=n;i++)
    {
        in>>v[i];
        update(i,v[i]);
    }
    while(q--)
    {
        in>>t;
        if(t==0)
        {
            in>>a>>b;
            update(a,b);
        }
        if(t==1)
        {
            in>>a>>b;
            out<<query(b)-query(a-1)<<"\n";
        }
        if(t==2)
        {
            in>>a;
            int st=1,dr=n,mij;
            while(st<dr)
            {
                mij=(st+dr)/2;
                if(query(mij)>=a)
                    dr=mij;
                else
                    st=mij+1;
            }
            if(query(st)!=a)
                out<<"-1\n";
            else
                out<<st<<"\n";
        }
    }
    return 0;
}