Cod sursa(job #2883957)

Utilizator T1raduTaerel Radu Nicolae T1radu Data 2 aprilie 2022 00:43:38
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <fstream>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n,m,aib[100001],v[100001];
int ub(int x)
{
    return x&(-x);
}
void add(int x, int val)
{
    for(int i=x;i<=n;i+=ub(i)) aib[i]+=val;
}
int sum(int x)
{
    int sum=0;
    for(int i=x;i>0;i-=ub(i)) sum+=aib[i];
    return sum;
}
int bs(int x)
{
    int st=1,dr=n;
    while(st<=dr)
    {
        int mid=(st+dr)/2;
        int T=sum(mid);
        if(T<=x) st=mid+1;
        else dr=mid-1;
    }
    return st-1;
}
int main()
{
    fin >> n >> m;
    for(int i=1;i<=n;i++)
    {
        fin >> v[i];
        add(i,v[i]);
    }
    for(int i=1;i<=m;i++)
    {
        int op,x,y;
        fin >> op;
        if(op==0)
        {
            fin >> x >> y;
            add(x,y);
        }
        else if(op==1)
        {
            fin >> x >> y;
            fout << sum(y)-sum(x-1) << "\n";
        }
        else
        {
            fin >> x;
            fout << bs(x) << "\n";
        }
    }
    return 0;
}