Cod sursa(job #2925429)

Utilizator Diana_IonitaIonita Diana Diana_Ionita Data 15 octombrie 2022 11:11:10
Problema Arbori indexati binar Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.99 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 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
        {int j;
            for( j=1; j<=n; j++)
            {
                if(interogare(j)==x) break;
            }
            if(j<=n)fout<<j<<'\n';
            else fout<<-1<<'\n';

        }
    }
    return 0;
}