Cod sursa(job #2948224)

Utilizator Zed1YasuoAlex Birsan Zed1Yasuo Data 27 noiembrie 2022 14:43:42
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <fstream>

using namespace std;
ifstream f("aib.in");
ofstream g("aib.out");
int n,m,a[100003],x,st,c,y,dr,mj;
void update(int poz,int x)
{
    for(int i=poz;i<=n;i+=(i&-i))
        a[i]+=x;
}
int query(int poz)
{
    int s=0;
    for(int i=poz;i;i-=(i&-i))
        s+=a[i];
    return s;
}
void sol(int x,int y)
{
    g<<query(y)-query(x-1)<<'\n';
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        f>>x;
        update(i,x);
    }
    for(int t=1;t<=m;t++)
    {
        f>>c;
        if(c==0)
        {
            f>>x>>y;
            update(x,y);
        }
        else
        if(c==1)
        {
            f>>x>>y;
            sol(x,y);
        }
        else
        {
            f>>x;
            st=0;dr=n;
            while(st<=dr)
            {
                mj=(st+dr)/2;
                int w=query(mj);
                if(w>x)
                    dr=mj-1;
                else
                    st=mj+1;
            }
            g<<st-1<<'\n';
        }
    }
    return 0;
}