Cod sursa(job #1856456)

Utilizator eduardturtoiEduard Turtoi eduardturtoi Data 24 ianuarie 2017 21:46:22
Problema Arbori indexati binar Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.32 kb
#include <fstream>
#define nmax 100001
#define zero(x) ((x ^ (x - 1)) & x)
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int aib[nmax],n;

void adauga(int poz,int nr)
{
    int i;
    for(i=poz;i<=n;i+=zero(i))
        aib[i]+=nr;
}

int calculeaza(int poz)
{
    int i,sum=0;
    for(i=poz;i>0;i-=zero(i))
        sum+=aib[i];
    return sum;
}

int cbin(int st,int dr,int val)
{
    int mij=(st+dr)/2;
    if(val==calculeaza(mij))
        return mij;
    else
    {
        if(st<dr)
        {
            if(val<calculeaza(mij))
                cbin(st,mij-1,val);
            else
                cbin(mij+1,dr,val);
        }
        else
            return -1;
    }
}

int main()
{
    int i,m,nr,op,poz,st,dr;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>nr;
        adauga(i,nr);
    }
    for(i=1;i<=m;i++)
    {
        fin>>op;
        if(op==0)
        {
            fin>>poz>>nr;
            adauga(poz,nr);
        }
        if(op==1)
        {
            fin>>st>>dr;
            fout<<calculeaza(dr)-calculeaza(st-1)<<'\n';
        }
        if(op==2)
        {
            fin>>nr;
            if(nr==0)
                fout<<0<<'\n';
            else
                fout<<cbin(1,n,nr)<<'\n';
        }
    }
    return 0;
}