Cod sursa(job #2617769)

Utilizator levladiatorDragutoiu Vlad-Ioan levladiator Data 22 mai 2020 20:17:37
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.22 kb
#include <bits/stdc++.h>
#define zeros(x) x&(-x)
#define NMAX 100005
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");

int n,m,put;
int aib[NMAX];

void update(int x,int val)
{
    for(int i=x;i<=n;i+=zeros(i))
        aib[i]+=val;
}
int compute(int x)
{
    int suma=0;
    for(int i=x;i>=1;i-=zeros(i))
        suma+=aib[i];

    return suma;
}
int cb(int x)
{
    int st=0;
    if(x==0)return -1;
    for(int dr=put;dr>=1;dr>>=1)
    {
        if(st+dr<=n)
        {
            if(aib[st+dr]<=x)
            {
                x-=aib[st+dr];
                st+=dr;
            }
            if(x==0)return st;
        }
    }
    return -1;
}
int main()
{
    fin>>n>>m;
    put=1;
    while(put<=n)
    {
        put<<=1;
    }
    put>>=1;

    for(int i=1;i<=n;i++)
    {
        int x;
        fin>>x;
        update(i,x);
    }
    for(int q=1;q<=m;q++)
    {
        int tip,a,b;
        fin>>tip;
        if(tip!=2)
        {
            fin>>a>>b;
            if(tip==0)update(a,b);
            else fout<<compute(b)-compute(a-1)<<'\n';
        }
        else
        {
            fin>>a;
            fout<<cb(a)<<'\n';
        }
    }
}