Cod sursa(job #3296333)

Utilizator iordacheMatei Iordache iordache Data 12 mai 2025 12:03:39
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <bits/stdc++.h>
#define pb push_back
//#define int long long
using namespace std;
const int N=1e5+5;
int n;
int aib[N];
void update(int i, int x)
{
    while(i<=n)
    {
        aib[i]+=x;
        i+=i&(-i);
    }
}
int pref(int i)
{
    int ans=0;
    while(i>0)
    {
        ans+=aib[i];
        i-=i&(-i);
    }
    return ans;
}
int query(int l, int r) {return pref(r)-pref(l-1);}
signed main()
{
    ifstream cin("aib.in");ofstream cout("aib.out");
    int q;
    cin>>n>>q;
    for(int i=1;i<=n;++i)
    {
        int x;cin>>x;
        update(i,x);
    }
    while(q--)
    {
        int op;cin>>op;
        if(op==0)
        {
            int i,x;cin>>i>>x;
            update(i,x);
        }
        else if(op==1)
        {
            int l,r;cin>>l>>r;
            cout<<query(l,r)<<'\n';
        }
        else
        {
            int x;cin>>x;
            int l=1,r=n,ans=0;
            while(l<=r)
            {
                int mij=(l+r)/2;
                if(pref(mij)<=x) ans=mij,l=mij+1;
                else r=mij-1;
            }
            if(ans==0||pref(ans)!=x) cout<<"-1\n";
            else cout<<ans<<'\n';
        }
    }
}