Cod sursa(job #2925433)

Utilizator Diana_IonitaIonita Diana Diana_Ionita Data 15 octombrie 2022 11:19:10
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.25 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int i,s,n,m,a,type,x,y,aib[100005];
void update(int zi,int val)
{
    int i;
    for(i=zi; i<=n; i+=i&(-i))
    {
        aib[i]+=val;
    }
}
int interogare(int x)
{
    s=0;
    int i;

    for(i=x; i>=1; i-=i&(-i))
    {
        s+=aib[i];
    }
    return s;

}


int cb(int val)
{
    int st = 1, dr = n, poz = -1;
    while (st <= dr)
    {
        int mid = (st + dr) / 2;
        int z = interogare(mid);

        if (z> val)
        {
            dr = mid - 1;
        }
        else if (z== val)
        {
            dr = mid - 1;
            poz = mid;
        }
        else
        {
            st = mid + 1;
        }
    }

    return poz;
}
int main()
{
    fin>>n>>m;
    for(i=1; i<=n; i++)
    {
        fin>>a;
        update(i,a);
    }
    for(i=1; i<=m; i++)
    {
        fin>>type>>x;
        if(type==0)
        {
            fin>>y;
            update(x,y);
        }
        else if(type==1)
        {
            fin>>y;
            fout<<interogare(y)-interogare(x-1)<<'\n';
        }
        else
        {
         fout<<cb(x)<<'\n';

        }
    }
    return 0;
}