Cod sursa(job #2914061)

Utilizator cyg_dragos10Ivan Dragos cyg_dragos10 Data 18 iulie 2022 15:53:56
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 aib[100005],n;

void update(int poz,int val)
{
    for(int i = poz;i <= n;i += i & -i)
        aib[i] += val;
}

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

int bs(int a)
{
    int st = 1,dr = n,mid;
    while(st <= dr)
    {
        mid = st + (dr - st) / 2;
        if(query(mid) < a)
            st = mid + 1;
        else
            dr = mid - 1;
    }
    if(query(st) == a)
        return st;
    return -1;

}
int main()
{
    int q,x,a,b;
    cin>>n>>q;
    for(int i = 1;i <= n;i++)
    {
        cin>>x;
        update(i,x);
    }
    for(int i = 1;i <= q;i++)
    {
        cin>>x;
        if(x == 0)
        {
            cin>>a>>b;
            update(a,b);
        }
        else
            if(x == 1)
            {
                cin>>a>>b;
                cout<<query(b) - query(a - 1)<<"\n";
            }
            else
            {
                cin>>a;
                cout<<bs(a)<<"\n";
            }
    }
    return 0;
}