Cod sursa(job #3277857)

Utilizator tudorhTudor Horobeanu tudorh Data 17 februarie 2025 16:11:48
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.86 kb
#include <bits/stdc++.h>

using namespace std;
#define zeros(x) ((x ^(x-1)) & x)
ifstream fin("aib.in");
ofstream fout("aib.out");
int aint[400001],a[100001];
long long sum;
int ans;
void build(int v,int st,int dr)
{
    if(st==dr)
        aint[v]=a[st];
    else
    {
        int mid=(st+dr)/2;
        build(v*2,st,mid);
        build(v*2+1,mid+1,dr);
        aint[v]=aint[v*2]+aint[v*2+1];
    }
}
void update(int v,int st,int dr,int pos,int val)
{
    if(st==dr)
        aint[v]+=val;
    else
    {
        int mid=(st+dr)/2;
        if(pos<=mid)
            update(v*2,st,mid,pos,val);
        else update(v*2+1,mid+1,dr,pos,val);
        aint[v]=aint[v*2]+aint[v*2+1];
    }
}
void querysum(int v,int st,int dr,int qst,int qdr)
{
    if(qst<=st && qdr>=dr)
        sum+=aint[v];
    else
    {
        int mid=(st+dr)/2;
        if(qst<=mid)
            querysum(v*2,st,mid,qst,qdr);
        if(qdr>mid)
            querysum(v*2+1,mid+1,dr,qst,qdr);
    }
}
void querypos(int v,int st,int dr,int val)
{
    if(st==dr)
    {
        if(aint[v]==val)
            ans=dr;
        else ans=-1;
    }
    else
    {
        int mid=(st+dr)/2;
        if(val<=aint[v*2])
            querypos(v*2,st,mid,val);
        else querypos(v*2+1,mid+1,dr,val-aint[v*2]);
    }
}
int main()
{
    int n,m,t,st,dr;
    fin>>n>>m;
    for(int i=1;i<=n;i++)
        fin>>a[i];
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        fin>>t;
        if(t==0)
        {
            fin>>st>>dr;
            update(1,1,n,st,dr);
        }
        else if(t==1)
        {
            fin>>st>>dr;
            sum=0;
            querysum(1,1,n,st,dr);
            fout<<sum<<'\n';
        }
        else
        {
            fin>>st;
            querypos(1,1,n,st);
            fout<<ans<<'\n';
        }
    }
    return 0;
}