Cod sursa(job #2775281)

Utilizator popescuadrianpopescuadrian popescuadrian Data 15 septembrie 2021 10:58:05
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <fstream>

using namespace std;
ifstream cin ("aib.in");
ofstream cout ("aib.out");
long long aib[150005],n;
void update (int poz,int val)
{
    while(poz<=n)
    {
        aib[poz]=aib[poz]+val;
        poz=poz+(poz& -poz);
    }
}
int qu (int poz)
{
    long long suma=0;
    while(poz)
    {
        suma=suma+aib[poz];
        poz=poz-(poz& -poz);
    }
    return suma;
}
int caut (int k)
{
    long long st=1,dr=n,ras,mij;
    while(st<=dr)
    {
        mij=(st+dr)/2;
        ras=qu(mij);
        if(ras>k)
        {
            dr=mij-1;
        }
        else
        {
            if(ras<k)
            {
                st=mij+1;
            }
            else
            {
                return mij;
            }
        }
    }
    return -1;
}
int main()
{
   long long i,j,x,q,cer,a,b,k;
    cin>>n>>q;
    for(i=1; i<=n; i++)
    {
        cin>>x;
        update(i,x);
    }
    for(i=1; i<=q; i++)
    {
        cin>>cer;
        if(cer==0)
        {
            cin>>a>>b;
            update(a,b);
        }
        else if(cer==1)
        {
            cin>>a>>b;
            cout<<qu(b)-qu(a-1)<<'\n';
        }
        else
        {
            cin>>k;
            cout<<caut(k)<<'\n';
        }
    }
    return 0;
}