Cod sursa(job #1268908)

Utilizator bulbulicaAlexandrescu Cristian bulbulica Data 21 noiembrie 2014 17:29:43
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.83 kb
#include <fstream>
#define Nmax 60200
using namespace std;
ifstream f("aib.in");
ofstream g("aib.out");

int vector[Nmax],n,m,indexare=1;

void actualizare(int nod,int st, int dr,int val,int poz)
{
    if (st==dr)
    {
        if (indexare)
            vector[nod]=val;
        else
            vector[nod]+=val;
        return;
    }
    int mij=(st+dr)/2;
    if (poz<=mij)
        actualizare(2*nod,st,mij,val,poz);
    else
        actualizare(2*nod+1,mij+1,dr,val,poz);
    vector[nod]=vector[2*nod]+vector[2*nod+1];
}
void interogare(int nod,int st,int dr,int inceput,int final,int &max)
{
    if (inceput<=st && final>=dr)
    {
        max+=vector[nod];
        return;
    }
    int mij =(st+dr)/2;
    if (inceput<=mij)
        interogare(2*nod,st,mij,inceput,final,max);
    if (mij<final)
        interogare(2*nod+1,mij+1,dr,inceput,final,max);
}
int main()
{
    int nr,operatie,val1,val2,suma,pozitie,valoare,k;
    f>>n>>m;
    for (int i=1;i<=n;++i)
    {
        f>>nr;
        pozitie=i;
        valoare=nr;
        actualizare(1,1,n,valoare,pozitie);
    }
    indexare=0;
    for (int i=1;i<=m;++i)
    {
        f>>operatie>>val1;
        if (operatie==0)
        {
            f>>val2;
            pozitie=val1;
            valoare=val2;
            actualizare(1,1,n,valoare,pozitie);
        }
        else if (operatie==2)
        {
            for (k=1;k<=n;++k)
            {
                suma=0;
                interogare(1,1,n,1,k,suma);
                if (suma==val1)
                {
                    g<<k<<'\n';
                    break;
                }
            }
        }
        else
        {
            f>>val2;
            suma=0;
            interogare(1,1,n,val1,val2,suma);
            g<<suma<<'\n';
        }
    }
    return 0;
}