Cod sursa(job #2409984)

Utilizator AndreiD31Dragan Andrei AndreiD31 Data 19 aprilie 2019 16:55:16
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.08 kb
#include <bits/stdc++.h>
#define indice(x) (x&(-x))

using namespace std;

ifstream f("aib.in");
ofstream g("aib.out");


int n,AIB[100100];


void adaug(int poz,int val)
{
    int i;
    for(i=poz;i<=n;i+=indice(i))
        AIB[i]+=val;
}

int suma(int poz)
{
    int i,suma=0;
    for(i=poz;i>=1;i=i-indice(i))
        suma+=AIB[i];
    return suma;
}



int m,x,mij,i,caz,copie,a,b,p,u;
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        adaug(i,x);
    }

    for(i=1;i<=m;i++)
    {
        f>>caz>>a;
        if(caz==0){f>>b; adaug(a,b);}
        else if(caz==1){f>>b; g<<suma(b)-suma(a-1)<<'\n';}
        else
        {
            p=1;u=n;copie=0;
            while(p<=u)
            {
                mij=(p+u)/2;

                if(suma(mij)>=a){ copie=mij; u=mij-1;}
                else if(suma(mij)<a)
                {
                    p=mij+1;
                }
            }




            if(suma(copie)==a)g<<copie<<'\n';
            else g<<-1<<'\n';

        }
    }

    return 0;
}