Cod sursa(job #3171821)

Utilizator Giulian617Buzatu Giulian Giulian617 Data 19 noiembrie 2023 17:02:15
Problema Arbori indexati binar Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("aib.in");
ofstream g("aib.out");
const int NMAX=100005;
int n,m,op,x,y;
vector<int>aib;
void update(int node,int value)
{
    for(int i=node;i<=n;i+=i&(-i))
        aib[i]+=value;
}
int query(int node)
{
    int ans=0;
    for(int i=node;i>0;i-=i&(-i))
        ans+=aib[i];
    return ans;
}
int magical_binary_search(int value)
{
    int ans=0,curr_sum=0;
    for(int step=(1<<30);step>0;step>>=1)
        if(ans+step<=n && curr_sum+aib[ans+step]<=value)
        {
            ans+=step;
            curr_sum+=aib[ans];
        }
    if(curr_sum==value)
        return ans;
    else
        return -1;
}
int main()
{
    f>>n>>m;
    aib.resize(n+1);
    for(int i=1;i<=n;i++)
    {
        f>>x;
        update(i,x);
    }
    for(int i=1;i<=m;i++)
    {
        f>>op;
        if(op==0)
        {
            f>>x>>y;
            update(x,y);
        }
        else if(op==1)
        {
            f>>x>>y;
            g<<query(y)-query(x-1)<<'\n';
        }
        else
        {
            f>>x;
            g<<magical_binary_search(x)<<'\n';
        }
    }
    return 0;
}