Cod sursa(job #3037684)

Utilizator vlad_lupuletiLupuleti Vlad Calin vlad_lupuleti Data 26 martie 2023 09:45:27
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.29 kb
#include<fstream>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int a,b,c,n,m,x,d[1000001];
int query(int p)
{
    int suma = 0;
    for (int i = p; i > 0; i-=(i&-i))
        suma += d[i];
    return suma;
}
void update(int p,int x)
{
    for (int i = p; i <= n; i+=(i&-i))
        d[i] += x;
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>x;
        update(i,x);
    }
    for(int i=1;i<=m;i++)
    {
        fin>>c>>a;
        if(c==0)
        {
            fin>>b;
            update(a,b);
        }
        if(c==1)
        {
            fin>>b;
            fout<<query(b)-query(a-1)<<"\n";
        }
        if(c==2)
        {
            int p1=1;
            int p2=n;
            int k=-1;
            while(p1<=p2)
            {
                int mij=(p1+p2)/2;
                int r=query(mij);
                if(r==a)
                {
                    k=mij;
                    p2=mij-1;
                }
                else
                {
                    if(r<a)
                    {
                        p1=mij+1;
                    }
                    else
                    {
                        p2=mij-1;
                    }
                }
            }
            fout<<k<<'\n';
        }
    }
    return 0;
}