Cod sursa(job #2376334)

Utilizator AndreiD31Dragan Andrei AndreiD31 Data 8 martie 2019 14:58:21
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.7 kb
#include <bits/stdc++.h>

using namespace std;

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

int v[100100];

void update(int nod,int st,int dr,int poz,int val)
{

    if(st==dr)v[nod]+=val;
    else
    {
        int m=(st+dr)/2;
        if(poz<=m)update(2*nod,st,m,poz,val);
        else update(2*nod+1,m+1,dr,poz,val);

        v[nod]=v[2*nod]+v[2*nod+1];
    }
}


int query(int nod,int st,int dr,int a,int b)
{
    //cout<<nod<<'\n';
    if(st>=a && dr<=b)return v[nod];
    else
    {
        int m=(st+dr)/2,k=0;
        if(a<=m)k+=query(2*nod,st,m,a,b);
        if(b>=m+1)k+=query(2*nod+1,m+1,dr,a,b);
        return k;
    }
}


int n,m,i,p,u,cp,mij,suma,val,c,a,b;

int main()
{
    f>>n>>m;
    for(i=1; i<=n; i++)
    {
        f>>val;
        update(1,1,n,i,val);
    }

    for(i=1; i<=m; i++)
    {
        f>>c;
        if(c==0)
        {
            f>>a>>b;
            update(1,1,n,a,b);
        }
        else if(c==1)
        {
            f>>a>>b;
            g<<query(1,1,n,a,b)<<'\n';
        }
        else
        {
            f>>a;

            p=1;
            u=n;
            cp=-1;
            while(p<=u)
            {
                mij=(p+u)/2;
                suma=query(1,1,n,1,mij);
                if(suma>a)
                {
                    u=mij-1;
                }
                else
                {
                    cp=mij;
                    p=mij+1;
                }
            }

            if(cp!=-1)
            {
                suma=query(1,1,n,1,cp);
                if(suma==a)g<<cp<<'\n';
            }
            else g<<-1<<'\n';
        }
    }


    return 0;
}