Cod sursa(job #2947040)

Utilizator Chris_BlackBlaga Cristian Chris_Black Data 25 noiembrie 2022 16:53:20
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <fstream>
#include <vector>
#define lsb(x) (x&(-x))
using namespace std;
ifstream cin("aib.in");
ofstream cout("aib.out");

vector<int> aib;
int n,m;
void Update(int poz,int val)
{
    for(int i=poz;i<=n;i += lsb(i))
        aib[i]+=val;
}
long long Compute(int poz)
{
    long long sum = 0;
    for(int i=poz;i>0; i-=lsb(i))
        sum += aib[i];
    return sum;
}

int cb(int val)
{
    int st = 1, dr = n, mij;
    while(st<=dr)
    {
        mij = (st+dr)/2;
        if(val==aib[mij])
            return mij;
        else
            if(val>aib[mij])
                st = mij+1;
            else dr = mij-1;
    }
    return -1;
}


int main()
{
    int x,cer,a,b;
    cin>>n>>m;
    aib = vector<int>(n+1);
    for(int i=1;i<=n;++i)
    {
        cin>>x;
        Update(i,x);
    }
    for(int k=1;k<=m;++k)
    {
        cin>>cer;
        if(cer==0)
        {
            cin>>a>>b;
            Update(a,b);
        }
        if(cer==1)
        {
            cin>>a>>b;
            cout<<Compute(b)-Compute(a-1)<<'\n';
        }
        else if(cer==2)
        {
            cin>>a;
            cout<<cb(a)<<'\n';
        }
    }
    return 0;
}