Cod sursa(job #2417285)

Utilizator Cezar211Popoveniuc Cezar Cezar211 Data 29 aprilie 2019 14:04:16
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.22 kb
#include <bits/stdc++.h>
#define NMAX 100005
using namespace std;
ifstream fin ("aib.in");
ofstream fout ("aib.out");
int n, m, v[NMAX];
void read();
void solve();
int lsb(int pos)
{
    return pos&-pos;
}
void update(int pos, int val)
{
    while(pos<=n)
    {
        v[pos]+=val;
        pos+=lsb(pos);
    }
}
int query(int pos)
{
    int s = 0;
    while(pos > 0)
    {
        s+=v[pos];
        pos-=lsb(pos);
    }
    return s;
}
int q3(int a)
{
    int st = 1, dr = n;
    while(st<=dr)
    {
        int mijl = (st+dr)/2, cur = query(mijl);
        if(cur == a)
            return mijl;
        else if(cur < a)
            st = mijl+1;
        else dr = mijl-1;
    }
    return -1;
}
int main()
{
    read();
    solve();
    return 0;
}
void read()
{
    int x;
    fin >> n >> m;
    for(int i=1; i<=n; i++)
    {
        fin >> x;
        update(i, x);
    }
}
void solve()
{
    int op, a, b;
    for(int q=1; q<=m; ++q)
    {
        fin >> op >> a;
        if(op!=2)
            fin >> b;
        if(op == 0)
            update(a, b);
        else if(op == 1)
            fout << query(b) - query(a-1) << '\n';
        else fout << q3(a) << '\n';
    }
}