Cod sursa(job #3366815)

Utilizator Moisiuc_MateiMoisiuc Matei Moisiuc_Matei Data 4 octombrie 2026 14:43:59
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.94 kb
#include <bits/stdc++.h>

using namespace std;

ifstream in("arbint.in");
ofstream out("arbint.out");

int t[500005];

void update(int nod,int st,int dr,int nr,int pos)
{
    if(dr==st)
    {
        t[nod]+=nr;
        return;
    }
    int m=(st+dr)/2;
    if(pos<=m)
    {
        update(2*nod,st,m,nr,pos);
    }
    else
    {
        update(2*nod+1,m+1,dr,nr,pos);
    }
    t[nod]=t[nod*2]+t[nod*2+1];
}
int s;
void intrebare(int nod,int st,int dr,int stdorit,int drdorit)
{
    if(stdorit<=st && dr<=drdorit)
    {
        s+=t[nod];
        return;
    }
    int m=(st+dr)/2;
    if(stdorit<=m)
    {
        intrebare(2*nod,st,m,stdorit,drdorit);
    }
    if(m<drdorit)
    {
        intrebare(2*nod+1,m+1,dr,stdorit,drdorit);
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n,q;
    cin>>n>>q;
    int nr;
    for(int i=1;i<=n;i++)
    {
        cin>>nr;
        update(1,1,n,nr,i);
    }
    int tip,a,b;
    for(int i=1;i<=q;i++)
    {
        cin>>tip;
        if(tip==1)
        {
            cin>>a>>b;
            s=0;
            intrebare(1,1,n,a,b);
            cout<<s<<"\n";
        }
        else if(tip==0)
        {
            cin>>a>>b;
            update(1,1,n,b,a);
        }
        else
        {
            int nr;
            cin>>nr;
            int dr=n,st=1,m,poz=-1,sumt=0;
            while(dr>=st)
            {
                m=(dr+st)/2;
                s=0;
                intrebare(1,1,n,st,m);
                if(s+sumt==nr)
                {
                    poz=m;
                    break;
                }
                if(s+sumt>nr)
                {
                    dr=m-1;
                }
                else
                {
                    st=m+1;
                    sumt+=s;
                }
            }
            cout<<poz<<"\n";
        }
    }
    return 0;
}