Cod sursa(job #3340713)

Utilizator BidonTurtitBezdedan Eric BidonTurtit Data 15 februarie 2026 21:45:22
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <iostream>
#include <fstream>
#define Max 100005
using namespace std;

ifstream fin("aib.in");
ofstream fout("aib.out");

int a[Max],n,m;
void citire()
{
    for(int i=1;i<=n;i++)
    {
        int x,p;
        fin>>x;
        a[i]+=x;
        p=i+(i&(-i));
        if(p<=n)
            a[p]+=a[i];
    }
}
void update(int pos, int val)
{
    while(pos<=n)
    {
        a[pos]+=val;
        pos=pos+(pos&(-pos));
    }
}

int query(int pos)
{
    int sum=0;
    while(pos>0)
    {
        sum+=a[pos];
        pos=pos-(pos&(-pos));
    }
    return sum;
}
int mn(int sum)
{
    int step;
    for(step=1;step<n;step=step*2);
    for(int i=0;step;step/=2)
    {
        if(i+step<=n)
        {
            if(sum>=a[i+step])
            {
                i+=step;
                sum-=a[i];
                if(sum==0)
                    return i;
            }
        }
    }
    return -1;
}
int main()
{
    fin>>n>>m;
    citire();
    for(int i=1;i<=m;i++)
    {
        int c,a,b;
        fin>>c;
        if(c==0)
        {
            fin>>a>>b;
            update(a,b);
        }
        if(c==1)
        {
            fin>>a>>b;
            fout<<query(b)-query(a-1)<<"\n";
        }
        if(c==2)
        {
            fin>>a;
            fout<<mn(a)<<"\n";
        }
    }
    return 0;
}