Cod sursa(job #2507826)

Utilizator AnduebossAlexandru Ariton Andueboss Data 10 decembrie 2019 21:46:29
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.08 kb
#include <fstream>
#define bits(i) (i&(-i))
using namespace std;
 
ifstream f("aib.in");
ofstream g("aib.out");
 
int n,aib[100100];
 
void adaug(int poz,int val)
{
    for(int i=poz;i<=n;i+=bits(i))
        aib[i]+=val;
}
 
int suma(int poz)
{
    int S=0;
    for(int i=poz;i>=1;i-=bits(i))
        S+=aib[i];
    return S;
}
 
 
int m,cerinta,i,x,a,b,cpy_poz,p,u,mij;
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        adaug(i,x);
    }
 
    for(int k=1;k<=m;k++)
    {
        f>>cerinta;
        if(cerinta==1)
        {
            f>>a>>b;
            g<<suma(b)-suma(a-1)<<'\n';
        }
        else if(cerinta==2)
        {
            f>>a;
            p=1;u=n;
            while(p<=u)
            {
                mij=(p+u)/2;
                if(suma(mij)>a){u=mij-1;}
                else {cpy_poz=mij; p=mij+1;}
            }
            //g<<cpy_poz<<" "<<suma(cpy_poz)<<"  ";
            if(suma(cpy_poz)==a)g<<cpy_poz<<'\n';
            else g<<-1<<'\n';
        }
        else
        {
            f>>a>>b;
            adaug(a,b);
        }
    }
    return 0;
}