Cod sursa(job #2869455)

Utilizator cdenisCovei Denis cdenis Data 11 martie 2022 15:43:43
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.02 kb
#include <iostream>
#include <fstream>

using namespace std;

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

const int MAX=1e5+5;
int n,m,t[MAX],a,b,op,val;

void adauga(int i, int val)
{
    for(;i<=n;i+=i&-i)
        t[i]+=val;
}

int sum(int i)
{
    int sum=0;
    for(;i;i-=i&-i)
        sum+=t[i];
    return sum;
}

int cb(int a)
{
    int st=1,dr=n;
    while(st<=dr)
    {
        int m=(st+dr)/2;
        if(a<=t[m])
            dr=m-1;
        else
            st=m+1;
    }
    return st;
}

int main()
{
    fin >> n >> m;
    for(int i=1;i<=n;i++)
    {
        fin >> val;
        adauga(i,val);
    }
    while(m--)
    {
        fin >> op;
        if(op==0)
        {
            fin >> a >> b;
            adauga(a,b);
        }
        else if(op==1)
        {
            fin >> a >> b;
            fout << sum(b)-sum(a-1) << '\n';
        }
        else
        {
            fin >> a;
            fout << cb(a) << '\n';
        }
    }
    return 0;
}