Cod sursa(job #3158480)

Utilizator Bianca2507Negret Bianca Bianca2507 Data 18 octombrie 2023 20:04:59
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>

using namespace std;
ifstream cin("aib.in");
ofstream cout("aib.out");
int n,m,aib[100005],x,ce,a,b;
void update(int p,int x)
{
    for(int i=p;i<=n;i+=(i&-i))
        aib[i]+=x;

}
int query(int p)
{
    int s=0;
    for(int i=p;i>=1;i-=(i&-i))
        s+=aib[i];
    return s;

}
int cb(int sum)
{
    int st=1,dr=n,mid,poz=-1,aux;
    while(st<=dr)
    {
        mid=(st+dr)/2;
        aux=query(mid);
        if(aux==sum)
        {
            poz=mid;
            dr=mid-1;
        }
        else
            if(aux<sum)
                st=mid+1;
            else
                dr=mid-1;
    }
    return poz;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>x;
        update(i,x);
    }
    for(int i=1;i<=m;i++)
    {
        cin>>ce;
        if(ce==0)
        {
         cin>>a>>b;
         update(a,b);
        }
        else
            if(ce==1)
        {
            cin>>a>>b;
            cout<<query(b)-query(a-1)<<'\n';

        }
        else
            if(ce==2)
        {
            cin>>a;
            cout<<cb(a)<<'\n';
        }
    }
    return 0;
}