Cod sursa(job #1284358)

Utilizator hanganflorinHangan Florin hanganflorin Data 6 decembrie 2014 14:26:20
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.1 kb
#include <fstream>
using namespace std;

ifstream is("aib.in");
ofstream os("aib.out");

int n, m, aib[100001], op, a, b;

void Update(int p, int v)
{
    for ( int i = p; i <= n; i += i & -i )
        aib[i] += v;
}
int Query(int p)
{
    int s = 0;
    for ( int i = p; i; i -= i & -i )
        s += aib[i];
    return s;
}
int Bs(int s)
{
    int st = 1, dr = n, mid;
    while(st < dr )
    {
        mid = (dr+st)/2;
        if ( Query(mid) >= s )
            dr = mid;
        else
            st = mid+1;
    }
    return st;
}

int main()
{
    is >> n >> m;
    for ( int i = 1, nr; i <= n; ++i )
    {
        is >> nr;
        Update(i, nr);
    }
    while ( m-- )
    {
        is >> op;
        if ( op == 0 )
        {
            is >> a >> b;
            Update(a, b);
        }
        else if ( op == 1 )
        {
            is >> a >> b;
            os << Query(b) - Query(a-1) << '\n';
        }
        else
        {
            is >> a;
            os << Bs(a) << '\n';
        }

    }
    is.close();
    os.close();
    return 0;
}