Cod sursa(job #2991212)

Utilizator Danut200333Dumitru Daniel Danut200333 Data 9 martie 2023 10:45:40
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.95 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int aib[100000],n,m,t,x,a,b,st,dr,mid;
void adun(int x,int a)
{
    for(int i=x;i<=n;i+=(i&-i))aib[i]+=a;
}
int suma(int x)
{
    int s=0;
    for(int i=x;i>=1;i-=(i&-i))s+=aib[i];
    return s;
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>x;
        adun(i,x);
    }
    while(m)
    {
        m--;
        fin>>t;
        if(t==0)
        {
            fin>>a>>b;
            adun(a,b);
        }
        if(t==1)
        {
            fin>>a>>b;
            fout<<suma(b)-suma(a-1)<<'\n';
        }
        if(t==2)
        {
            fin>>a;
            st=1;dr=n;
            while(st<=dr)
            {
                mid=(st+dr)/2;
                if(suma(mid)>a)dr=mid-1;
                else st=mid+1;
            }
            fout<<st-1<<'\n';
        }
    }
    return 0;
}