Cod sursa(job #2071858)

Utilizator GeorgeCalinPetruta George-Calin GeorgeCalin Data 21 noiembrie 2017 08:52:51
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>
#define nmax 100002
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");

int arb[nmax];
int val,n,m,c,x,y;

void update(int poz)
{
    while(poz<=n)
    {
        arb[poz]+=val;
        poz+=(poz&-poz);
    }
}

int query(int poz)
{
    int s=0;
    while(poz)
    {
        s+=arb[poz];
        poz-=(poz&-poz);
    }
    return s;
}

int cautare(int val)
{
    int i,pas;
    for(pas=1;pas<=n;pas<<=1);
    for(i=0;pas;pas>>=1)
    {
        if(i+pas<=n)
        {
            if(val>=arb[i+pas])
            {
                i+=pas;
                val-=arb[i];
                if(!val) return i;
            }
        }
    }
    return -1;
}

int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>val;
        update(i);
    }
    for(int i=1;i<=m;i++)
    {
        fin>>c;
        if(c==0)
        {
            fin>>x>>y;
            val=y;
            update(x);
        }
        if(c==1)
        {
            fin>>x>>y;
            fout<<query(y)-query(x-1);
            fout<<"\n";
        }
        if(c==2)
        {
            fin>>x;
            fout<<cautare(x);
            fout<<"\n";
        }
    }
    return 0;
}